Ошибка в минимальном остовном дереве по алгоритму Прима - PullRequest
0 голосов
/ 17 апреля 2020

Я пытаюсь реализовать алгоритм Прима, но вывод меняется каждый раз, когда я его запускаю

{'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')))

1 Ответ

1 голос
/ 17 апреля 2020

Если присмотреться, оба выхода,

{'A': {'C'}, 'B': {'A'}, 'F': {'G'}, 'E': {'B'}, 'D': {'E'}, 'C': {'F'}} и {'D': {'E'}, 'E': {'B'}, 'B': {'A'}, 'A': {'C'}, 'C': {'F'}, 'F': {'G'}} одинаковы.

Так как это словарь; Имеют значение пары (ключ, значение), которые совпадают в обоих ответах, а не порядок их появления.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...