About Blog GitHub

15 May 2018
广度优先搜索算法(go)

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

本文通过go语言实现广度优先搜索算法,使用该算法从朋友圈中找出关系最近的售货员朋友。

下面介绍详细的实现过程。首先,调用createFriendCircle函数创建一个模拟的朋友圈,朋友圈如下图所示:

图1 参考《算法图解》

朋友圈通过字典实现,以“你”(you)作为根节点,字典的键是朋友圈属主,值是朋友圈所有朋友名字,通过一个列表方式实现,名字按字母顺序排序。

其次,传递创建的朋友圈给breadthFirstSearch函数,该函数是广度优先搜索算法的具体实现,在函数内部,首先取出you的所有朋友,如果朋友数为0,查找失败,返回false。如果朋友数不为0,则从you的所有朋友中取出一个朋友,并将朋友从待查找的朋友列表中删除,然后创建一个字典记录被查找过的朋友,避免再次查找。如果该朋友没有被检查过,则检查该朋友是否是售货员(名字以字母y结尾)。如果是售货员,查找成功,返回true。如果该朋友不是售货员,将该朋友的所有朋友又添加到待查找朋友列表中,继续查找,直到结束,实现一种类似Z字形的搜索路径。

由示例中可以看到,查找到的售货员是peggy,而不是jonny。因为这里的朋友名字是按字母顺序排序,所以优先查找了bob的朋友,而不是claire的朋友,即peggy是朋友圈中距离you最近的售货员朋友。

示例代码如下:

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!
*/

参考:



LEo at 22:32

About Blog GitHub