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"] = []
#起始顶点和结束顶点
start='you'
end='thom'
distance=0
#创建队列
search_queue = deque()
#往队列中加入起始顶点关联的下一批
search_queue+=graph[start]
#print search_queue
#标记已经遍历过的顶点
searched=[start]
#当队列不为空时
while search_queue:
#将队列第一位取出
person = search_queue.popleft()
#print person
#如果该顶点还没有被遍历
if person not in searched:
#且该顶点已经到达目的
if person == end:
searched.append(end)
print searched
#该顶点已经被遍历
else:
# 将此人关联的其他人加入队列
search_queue += graph[person]
searched.append(person)
#print search_queue
输出
['you', 'alice', 'bob', 'claire', 'peggy', 'anuj', 'thom']
但是这个只能得到顶点的遍历顺序(输出searched数组即可)。如果我们想将start=‘you’,
end='thom’这两个顶点间的路径和路径长度求出来,单单上面的代码是不够的。
#打印最短路径
l=[]
endd='thom'
l=[endd]
#没有找到其实顶点you
while endd != 'you':
#遍历searched的顶点
for i in searched:
#如果找到endd的父亲
if endd in graph[i]:
#将父亲加入到数组
l.append(i)
#令endd=父亲顶点(继续找父亲的父亲,直到找到起始顶点)
endd=i
break
print l
print l[::-1]
输出
['thom', 'claire', 'you']
['you', 'claire', 'thom']