|
|
// File: graph_bfs_test.go
|
|
|
// Created Time: 2023-02-18
|
|
|
// Author: Reanon (793584285@qq.com)
|
|
|
|
|
|
package chapter_graph
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"testing"
|
|
|
|
|
|
. "github.com/krahets/hello-algo/pkg"
|
|
|
)
|
|
|
|
|
|
func TestGraphBFS(t *testing.T) {
|
|
|
/* 初始化無向圖 */
|
|
|
vets := ValsToVets([]int{0, 1, 2, 3, 4, 5, 6, 7, 8, 9})
|
|
|
edges := [][]Vertex{
|
|
|
{vets[0], vets[1]}, {vets[0], vets[3]}, {vets[1], vets[2]}, {vets[1], vets[4]},
|
|
|
{vets[2], vets[5]}, {vets[3], vets[4]}, {vets[3], vets[6]}, {vets[4], vets[5]},
|
|
|
{vets[4], vets[7]}, {vets[5], vets[8]}, {vets[6], vets[7]}, {vets[7], vets[8]}}
|
|
|
graph := newGraphAdjList(edges)
|
|
|
fmt.Println("初始化後，圖為:")
|
|
|
graph.print()
|
|
|
|
|
|
/* 廣度優先走訪 */
|
|
|
res := graphBFS(graph, vets[0])
|
|
|
fmt.Println("廣度優先走訪（BFS）頂點序列為:")
|
|
|
PrintSlice(VetsToVals(res))
|
|
|
}
|
