广度优先搜索算法（go）

Posted 2021-04-14 LEo的网络日志

广度优先搜索算法（Breadth First Search，缩写为BFS），又译作宽度优先搜索，或横向优先搜索，是一种图形搜索算法。简单的说，广度优先搜索算法是从根节点开始，沿着树的宽度遍历树的节点。如果所有节点均被访问，则算法中止。借助广度优先搜索算法，可以让你找出两样东西之间的最短距离。

图1 参考《算法图解》

示例代码如下：

package main

import "fmt"

func main() {
    friendCircle := createFriendCircle()
    breadthFirstSearch(friendCircle)
}

func personIsSeller(name string) bool {
    return name[len(name)-1] == 'y'
}

func createFriendCircle() map[string][]string {
    fc := make(map[string][]string)
    fc["you"] = []string{"alice", "bob", "claire"}
    fc["bob"] = []string{"anuj", "peggy"}
    fc["alice"] = []string{"peggy"}
    fc["claire"] = []string{"thom", "jonny"}
    fc["anuj"] = []string{}
    fc["peggy"] = []string{}
    fc["thom"] = []string{}
    fc["jonny"] = []string{}
    return fc
}

func breadthFirstSearch(friendCircle map[string][]string) bool {
    searchList := friendCircle["you"]
    if len(searchList) == 0 {
        return false
    }

    searched := make(map[string]bool)
    for {
        person := searchList[0]
        searchList = searchList[1:]
        _, found := searched[person]
        if !found {
            if personIsSeller(person) {
                fmt.Println(person + " is a seller!")
                return true
            } else {
                searchList = append(searchList, friendCircle[person]...)
                searched[person] = true
            }
        }

        if len(searchList) == 0 {
            break
        }
    }

    return false
}

/*
peggy is a seller!
*/

参考：

• 《算法图解》

• wiki：广度优先搜索