基本上,我已经绞尽脑汁想知道为什么我的代码不起作用,我已经单独测试了各个部分,并在整个网络上查找它是否有帮助,但无济于事。 我收到一个错误,回溯是:
Traceback (most recent call last):
File "yes2.py", line 62, in <module>
g.add_edge(row_index,col_index, b)
File "yes2.py", line 27, in add_edge
self.adj[u].append(edge)
KeyError: 0
有错误的两部分是
def add_edge(self, u, v, w=0):
if u == v:
raise ValueError("u == v")
edge = Edge(u,v,w)
redge = Edge(v,u,0)
edge.redge = redge
redge.redge = edge
self.adj[u].append(edge) #### LINE 27 ####
self.adj[v].append(redge)
self.flow[edge] = 0
self.flow[redge] = 0
和
g = FlowNetwork()
map(g.add_vertex, ['0','1','2','3','4','5','6'])
with open('network.txt', "r") as file:
for row_index, row in enumerate(file):
for col_index, value in enumerate(row.split(",")):
b = int(value)
if b != 0:
g.add_edge(row_index,col_index, b) ### LINE 62 ####
这是完成的代码,因为没有这个代码可能很难看到发生了什么
class Edge(object):
def __init__(self, u, v, w):
self.source = u
self.sink = v
self.capacity = w
def __repr__(self):
return "%s->%s:%s" % (self.source, self.sink, self.capacity)
class FlowNetwork(object):
def __init__(self):
self.adj = {}
self.flow = {}
def add_vertex(self, vertex):
self.adj[vertex] = []
def get_edges(self, v):
return self.adj[v]
def add_edge(self, u, v, w=0):
if u == v:
raise ValueError("u == v")
edge = Edge(u,v,w)
redge = Edge(v,u,0)
edge.redge = redge
redge.redge = edge
self.adj[u].append(edge)
self.adj[v].append(redge)
self.flow[edge] = 0
self.flow[redge] = 0
def find_path(self, source, sink, path):
if source == sink:
return path
for edge in self.get_edges(source):
residual = edge.capacity - self.flow[edge]
if residual > 0 and not (edge,residual) in path:
result = self.find_path( edge.sink, sink, path + [(edge,residual)] )
if result != None:
return result
def max_flow(self, source, sink):
path = self.find_path(source, sink, [])
while path != None:
flow = min(res for edge,res in path)
for edge,res in path:
self.flow[edge] += flow
self.flow[edge.redge] -= flow
path = self.find_path(source, sink, [])
return sum(self.flow[edge] for edge in self.get_edges(source))
g = FlowNetwork()
map(g.add_vertex, ['0','1','2','3','4','5','6'])
with open('network.txt', "r") as file:
# enumerate allows you to iterate through the list with an index and an object
for row_index, row in enumerate(file):
# split allows you to break a string apart with a string key
for col_index, value in enumerate(row.split(",")):
#convert value from string to int
b = int(value)
if b != 0:
g.add_edge(row_index,col_index, b)
print g.max_flow('1','6')
非常感谢您抽出时间,非常感谢。
您收到的错误是
self.adj
还没有密钥 0
。您正在尝试追加到尚不存在的列表。
defaultdict
代替,替换此行(在 __init__
中):
self.adj = {}
这样:
self.adj = defaultdict(list)
您需要在顶部导入:
from collections import defaultdict
现在,
KeyError
不再提出 self.adj[0].append(edge)
,而是自动创建一个列表以追加到其中。
defaultdict 解决方案更好。 但为了完整起见,您还可以在追加之前检查并创建空列表。 添加 + 行:
+ if not u in self.adj.keys():
+ self.adj[u] = []
self.adj[u].append(edge)
.
.
仅当您的列表或词典在本地功能中不可用时才会出现。
试试这个:
class Flonetwork(Object):
def __init__(self,adj = {},flow={}):
self.adj = adj
self.flow = flow
当您尚未为键定义任何值并尝试使用
+=
运算符向其附加新值时,会发生此错误。例如。
s = "PA"
dict = {}
r = 0
for i in s:
dict[r] += i
这将导致 KeyError,因为我还没有为 Key 定义任何值
dict[r]
但是如果你写一段代码:
for j in range(len(s)):
dict[j] = ''
那么你将不会得到 KeyErrors,因为你已经将 dict 定义为:
{0: '', 1: ''}
希望这有帮助!