我正在尝试实现Prim的算法,但是每次运行它时输出都会改变
{'A': {'C'}, 'B': {'A'}, 'F': {'G'}, 'E': {'B'}, 'D': {'E'}, 'C': {'F'}}
应该是]
{'D': {'E'}, 'E': {'B'}, 'B': {'A'}, 'A': {'C'}, 'C': {'F'}, 'F': {'G'}}
[不确定这里到底发生了什么,我尝试调试无济于事。有人知道我是否缺少明显的东西吗?
from collections import defaultdict
import heapq
def create_spanning_tree(graph, starting_vertex):
mst = defaultdict(set)
visited = set([starting_vertex])
edges = [
(cost, starting_vertex, to)
for to, cost in graph[starting_vertex].items()
]
heapq.heapify(edges)
while edges:
cost, frm, to = heapq.heappop(edges)
if to not in visited:
visited.add(to)
mst[frm].add(to)
for to_next, cost in graph[to].items():
if to_next not in visited:
heapq.heappush(edges, (cost, to, to_next))
return mst
example_graph = {
'A': {'B': 2, 'C': 3},
'B': {'A': 2, 'C': 12, 'D': 10, 'E': 4},
'C': {'A': 3, 'B': 12, 'F': 5},
'D': {'B': 10, 'E': 7},
'E': {'B': 4, 'D': 7, 'F': 16},
'F': {'C': 5, 'E': 16, 'G': 9},
'G': {'F': 9},
}
print(dict(create_spanning_tree(example_graph, 'D')))
我正在尝试实现Prim的算法,但是每次运行它时输出都会更改{'A':{'C'},'B':{'A'},'F':{'G'},' E':{'B'},'D':{'E'},'C':{'F'}},当应为{'D':{'E'},'...
仔细观察,两个输出,