Алгоритм хопкрофта-карпа в Python

Я пытаюсь реализовать алгоритм Хопкрофта карпа в Python с использованием networkx в качестве графического представления.

В настоящее время я так далеко:

#Algorithms for bipartite graphs

import networkx as nx
import collections

class HopcroftKarp(object):
    INFINITY = -1

    def __init__(self, G):
        self.G = G

    def match(self):
        self.N1, self.N2 = self.partition()
        self.pair = {}
        self.dist = {}
        self.q = collections.deque()

        #init
        for v in self.G:
            self.pair[v] = None
            self.dist[v] = HopcroftKarp.INFINITY

        matching = 0

        while self.bfs():
            for v in self.N1:
                if self.pair[v] and self.dfs(v):
                    matching = matching + 1

        return matching

    def dfs(self, v):
        if v != None:
            for u in self.G.neighbors_iter(v):
                if self.dist[ self.pair[u] ] == self.dist[v] + 1 and self.dfs(self.pair[u]):
                    self.pair[u] = v
                    self.pair[v] = u

                    return True

            self.dist[v] = HopcroftKarp.INFINITY
            return False

        return True

    def bfs(self):
        for v in self.N1:
            if self.pair[v] == None:
                self.dist[v] = 0
                self.q.append(v)
            else:
                self.dist[v] = HopcroftKarp.INFINITY

        self.dist[None] = HopcroftKarp.INFINITY

        while len(self.q) > 0:
            v = self.q.pop()
            if v != None:
                for u in self.G.neighbors_iter(v):
                    if self.dist[ self.pair[u] ] == HopcroftKarp.INFINITY:
                        self.dist[ self.pair[u] ] = self.dist[v] + 1
                        self.q.append(self.pair[u])

        return self.dist[None] != HopcroftKarp.INFINITY


    def partition(self):
        return nx.bipartite_sets(self.G)

алгоритм взят из http://en.wikipedia.org/wiki/Hopcroft%E2%80%93Karp_algorithm Однако это не работает. Я использую следующий тестовый код

G = nx.Graph([
(1,"a"), (1,"c"),
(2,"a"), (2,"b"),
(3,"a"), (3,"c"),
(4,"d"), (4,"e"),(4,"f"),(4,"g"),
(5,"b"), (5,"c"),
(6,"c"), (6,"d")
])

matching = HopcroftKarp(G).match()

print matching

к сожалению, это не работает, я в конечном итоге в бесконечный цикл :(. Может кто-нибудь заметить ошибку, я вышел идей, и я должен признать, что я еще не полностью понимаю алгоритм, поэтому это в основном реализация псевдокода в Википедии

7
algorithm bipartite graph graph-algorithm python

2 ответов

строку 

if self.pair[v] and self.dfs(v):

должно быть

if self.pair[v] is None and self.dfs(v):

согласно псевдо-коду на странице Википедии. Единственная другая проблема, которую я вижу, заключается в том, что вы используете deque как стек, и вы хотите использовать его как очередь. Чтобы исправить это, вам просто нужно popleft, а не pop (который появляется справа). Итак, линия

v = self.q.pop()

должно быть

v = self.q.popleft()

надеюсь, все остальное работает. Я просто проверял, что ваш код Python работает так же, как и псевдокод в Википедии, надеюсь, что псевдокод правильный.

4

в Python есть пакет для этого алгоритма. HopcroftKarp, вы можете напрямую использовать этот пакет для своей реализации.

1