我正在尝试实现Karger 算法来查找图形的最小切割。关键部分是contract
执行单次收缩的方法。到目前为止,这是我的实现(带有“测试”):
import pytest
import random
class Graph(object):
def __init__(self, G):
self.G = G # Adjacency list
@property
def edges(self):
E = list()
for vertex in self.G:
for adjacent_vertex in self.G[vertex]:
if vertex < adjacent_vertex:
E.append([vertex, adjacent_vertex])
return E
def randomized_contract(self):
edge = random.choice(self.edges)
self.contract(edge)
def contract(self, edge):
vertex, adjacent_vertex = edge
self.G[vertex].remove(adjacent_vertex)
self.G[adjacent_vertex].remove(vertex)
self.G[vertex] += self.G[adjacent_vertex]
del self.G[adjacent_vertex]
for v in self.G:
for n, av in enumerate(self.G[v]):
if av == adjacent_vertex:
self.G[v][n] = vertex
self.remove_self_loops()
def remove_self_loops(self):
for vertex in self.G:
for n, adjacent_vertex in enumerate(self.G[vertex]):
if adjacent_vertex == vertex:
del self.G[vertex][n]
def contract_till_cut(self):
while len(self.G) > 2:
self.randomized_contract()
def test_contract_till_cut():
graph = Graph({1: [2,3], 2: [1,3], 3: [1,2,4], 4: [3]})
graph.contract_till_cut()
print(graph.G)
if __name__ == "__main__":
pytest.main([__file__, "-s"])
我的问题是,在一次特定的运行中(您可能需要运行几次才能重现此结果),获取邻接列表
{1: [1, 4], 4: [1]}
其中节点 1 有一个“自循环”——也就是说,它出现在自己的邻接列表中。我不明白这怎么会发生。每次调用都以似乎有效contract
的调用结束。remove_self_loops
谁能发现这段代码中的错误?