在Python中實施Kruskal算法時輸出錯誤

Question

我正在研究Krusal算法。 但是我沒有得到正確的輸出。 我不知道我在哪里做薄餅。

這是我的代碼：

parent = dict()
rank = dict()

def make_set(vertice):
    parent[vertice] = vertice
    rank[vertice] = 0

def find(vertice):
    if parent[vertice] != vertice:
        parent[vertice] = find(parent[vertice])
    return parent[vertice]

def union(vertice1, vertice2):
    root1 = find(vertice1)
    root2 = find(vertice2)
    if root1 != root2:
        if rank[root1] > rank[root2]:
            parent[root2] = root1
        else:
            parent[root1] = root2
            if rank[root1] == rank[root2]: rank[root2] += 1

def kruskal(graph):
    for vertice in graph['vertices']:
        make_set(vertice)

    minimum_spanning_tree = set()
    edges = list(graph['edges'])

    for edge in edges:
        vertice1, vertice2, weight = edge
        if find(vertice1) != find(vertice2):
            union(vertice1, vertice2)
            minimum_spanning_tree.add(edge)
    return minimum_spanning_tree






graph = {
    'vertices': [0,1, 2, 3, 4, 5],
    'edges': set([     //(first node, second node, weight)
        (0, 3,5),
        (3, 5,2),
        ( 5, 4,10),
        ( 4, 1,3),
        (1, 0,8),
        (0, 2,1),(2,3,6),( 2,5,4),( 2,4,9),(2,1,7),
        ])
    }

a = kruskal(graph)

print(a)

我發現，如果將“邊緣”更改為（權重，第一節點，第二節點）格式，則可以以（權重，第一節點，第二節點）格式得到正確答案。 但是，如果我想以（第一個節點，第二個節點，weigth）的格式使用“ Edges”，該如何修改？

Answer 1

乍一看，試試這個：

edges = list(graph['edges'])

代替

edges = sorted(graph['edges'], key=lambda e: e[2])