[Python] 纯文本查看 复制代码
from collections import deque
def do_map():
"""实现图"""
graph = dict()
graph["you"] = ["zhangsan", "lisi", "wangwu"]
graph["zhangsan"] = ["xiaoming", "xiaohong"]
graph["lisi"] = ["liniu", "you"]
graph["wangwu"] = ["wangniu", "mango"]
graph["xiaoming"] = []
graph["xiaohong"] = []
graph["liniu"] = []
graph["wangniu"] = []
graph["mango"] = []
return graph
def check_person(person):
"""检查是否是销售商"""
if person == 'mango': # 给一个简单的条件，只要名字是mango的就是销售商
return True
return False
def search(name, graph):
"""广度优先搜索-销售商"""
search_queue = deque() # 创建队列
search_queue += graph[name] # 将一度关系加入队列
searched = [] # 准备一个已经搜寻过销售商的列表
# 只要队列不为空，
while search_queue:
person = search_queue.popleft() # 取出第一个人
# 判断该人是否已经检查过了
if person not in searched:
# 判断该数据是否是要搜索的销售商
if check_person(person):
print('%s is a mango seller' % person)
return True
else:
# 不是，将这个人的朋友即二度关系加入队列
search_queue += graph[person]
# 将这个人标记为检查过
searched.append(person)
# 队列查完都没有就说明没找到
return False
if __name__ == '__main__':
graph = do_map()
print(search('you', graph))