from collections import deque

#解決從你的人際關系網(wǎng)中找到芒果銷售商的問題
#使用字典表示映射關系
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'

#實現(xiàn)廣度優(yōu)先搜索算法 
def search(name):
   search_queue = deque() #創(chuàng)建一個隊列
   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

#調(diào)用方法
search("you")