0

このきちんとしたクラスカルのアルゴリズムの実装を見つけたので、最小スパニング ツリーの代わりに最大スパニング ツリーを生成するように、それを「反転」させたいと思います。私は組合に変更しようとするのに十分なほど素朴でしたif rank[root1] > rank[root2]:. if rank[root1] < rank[root2]:明らかにこれはうまくいきませんでした。また、いくつかの場所で root1 と root2 を入れ替えてみました。どうやらこのコードの複雑さは私のスキルを超えています。コード:

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'])
        edges.sort()
#print edges
    for edge in edges:
        vertice1, vertice2, weight = edge
        if find(vertice1) != find(vertice2):
            union(vertice1, vertice2)
            minimum_spanning_tree.add(edge)

    return sorted(minimum_spanning_tree)

graph = {
'vertices': ['A', 'B', 'C', 'D', 'E', 'F', 'G'],
'edges': set([
    ('A', 'B', 7),
    ('A', 'D', 5),
    ('B', 'C', 8),
    ('B', 'D', 9),
    ('B', 'E', 7),
    ('C', 'E', 5),
    ('D', 'E', 15),
    ('D', 'F', 6),
    ('E', 'F', 8),
    ('E', 'G', 9),
    ('F', 'G', 11)
])
}

print(kruskal(graph))

最大スパニング ツリーを生成するようにスクリプトを変更するにはどうすればよいですか? 返されたグラフが

edges = [
    ('A', 'B', 7),
    ("A", "D", 5),
    ("B", "C", 8),
    ("B", "D", 9),
    ("B", "E", 7),
    ("C", "E", 5),
    ("D", "E", 15),
    ("D", "F", 6),
    ("E", "F", 8),
    ("E", "G", 9),
    ("F", "G", 11)
]

ありがとうございました!

4

0 に答える 0