hello-algo/ja/codes/go/chapter_graph/graph_bfs.go

// File: graph_bfs.go
// Created Time: 2023-02-18
// Author: Reanon (793584285@qq.com)

package chapter_graph

import (
	. "github.com/krahets/hello-algo/pkg"
)

/* 幅優先探索 */
// グラフを隣接リストで表し、指定した頂点の隣接頂点をすべて取得できるようにする
func graphBFS(g *graphAdjList, startVet Vertex) []Vertex {
	// 頂点の走査順序
	res := make([]Vertex, 0)
	// 訪問済み頂点を記録するためのハッシュ集合
	visited := make(map[Vertex]struct{})
	visited[startVet] = struct{}{}
	// キューは BFS の実装に用い、スライスでキューをシミュレートする
	queue := make([]Vertex, 0)
	queue = append(queue, startVet)
	// 頂点 vet を起点に、すべての頂点を訪問し終えるまで繰り返す
	for len(queue) > 0 {
		// 先頭の頂点をデキュー
		vet := queue[0]
		queue = queue[1:]
		// 訪問した頂点を記録
		res = append(res, vet)
		// この頂点のすべての隣接頂点を走査
		for _, adjVet := range g.adjList[vet] {
			_, isExist := visited[adjVet]
			// 未訪問の頂点のみをキューに追加
			if !isExist {
				queue = append(queue, adjVet)
				visited[adjVet] = struct{}{}
			}
		}
	}
	// 頂点の走査順を返す
	return res
}