python實現廣度優先搜索

from collections import deque

#解決從你的人際關係網中找到芒果銷售商的問題
#使用字典表示映射關係
graph = {}
graph["you"] = ["alice", "bob", "claire"]
graph["bob"] = ["anuj", "peggy"]
graph["alice"] = ["peggy"]
graph["claire"] = ["thom", "jonny"]
graph["anuj"] = []
graph["peggy"] = []
graph["thom"] = []
graph["jonny"] = []

#判斷是不是要查找的目標
def is_target_node(name):
return name[-1] == 'm'

#實現廣度優先搜索算法
def search(name):
search_queue = deque() #建立一個隊列
search_queue += graph[name]
searched = [] #記錄用於檢查過的人
while search_queue: #只要隊列不爲空
person = search_queue.popleft() #就取出其中的第一我的
if not person in searched: #這我的沒有被檢查過
if is_target_node(person): #判斷這我的是不是要查找的銷售商
print(person + " is target node!")
return True
else:
search_queue += graph[person] #若是這我的不是,就將這我的的朋友壓入隊列
searched.append(person) #將這我的追加到已檢查過的字典中
return False

#調用方法
search("you")
相關文章
相關標籤/搜索