基于python模擬bfs和dfs代碼實例
更新時間:2020年11月19日 10:43:43 作者:Neaya
這篇文章主要介紹了基于python模擬bfs和dfs代碼實例,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
BFS
"""
# @Time : 2020/11/8
# @Author : Jimou Chen
"""
# 廣搜
def bfs(graph, start):
queue = [start] # 先把起點入隊列
visited = set() # 訪問國的點加入
visited.add(start)
while len(queue):
vertex = queue.pop(0)
# 找到隊列首元素的連接點
for v in graph[vertex]:
if v not in visited:
queue.append(v)
visited.add(v)
# 打印彈出隊列的該頭元素
print(vertex, end=' ')
if __name__ == '__main__':
graph = {
'A': ['B', 'D', 'I'],
'B': ['A', 'F'],
'C': ['D', 'E', 'I'],
'D': ['A', 'C', 'F'],
'E': ['C', 'H'],
'F': ['B', 'H'],
'G': ['C', 'H'],
'H': ['E', 'F', 'G'],
'I': ['A', 'C']
}
bfs(graph, 'A')
A B D I F C H E G
Process finished with exit code 0
DFS
"""
# @Time : 2020/11/8
# @Author : Jimou Chen
"""
# 深搜
def dfs(graph, start):
stack = [start]
visited = set()
visited.add(start)
while len(stack):
vertex = stack.pop() # 找到棧頂元素
for v in graph[vertex]:
if v not in visited:
stack.append(v)
visited.add(v)
print(vertex, end=' ')
if __name__ == '__main__':
graph = {
'A': ['B', 'D', 'I'],
'B': ['A', 'F'],
'C': ['D', 'E', 'I'],
'D': ['A', 'C', 'F'],
'E': ['C', 'H'],
'F': ['B', 'H'],
'G': ['C', 'H'],
'H': ['E', 'F', 'G'],
'I': ['A', 'C']
}
dfs(graph, 'E')
E H G F B A I D C
Process finished with exit code 0
總結
很明顯一個用了隊列,一個用了棧
利用python語言優(yōu)勢,只需改動pop即可
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持腳本之家。
相關文章
pytorch numpy list類型之間的相互轉(zhuǎn)換實例
今天小編就為大家分享一篇pytorch numpy list類型之間的相互轉(zhuǎn)換實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08

