廣度優先搜索算法(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!
*/

參考:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章