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"] = []

#起始顶点和结束顶点
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']
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值