我在我的一个递归函数中遇到了内存错误。
def allPaths(self, adjMat, start, stop, flag=[0], walks=[]):
walks = walks + [start]
if start == stop:
return [walks]
loc = 0
flag=flag*len(adjMat)
output = []
for value in adjMat[start]:
if value > 0.0:
if flag[loc] < 3:
flag[loc]+=1
paths = self.allPaths(adjMat, loc, stop, flag, walks)
for k in paths:
output.append(k)
loc += 1
return output
一个示例输入很好,但是我得到了不同矩阵的内存错误。
>>>print test.allPaths([[0.0, 0.9, 0.0, 0.0, 0.0, 0.0, 0.1, 0.0],
[0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0],
[0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0],
[0.0, 0.0, 0.0, 0.8, .15, .05, 0.0, 0.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0]],0,7)
[[0, 1, 2, 3, 3, 3, 4, 7], [0, 1, 2, 3, 3, 3, 5, 7], [0, 1, 2, 3, 3, 4, 7], [0, 1, 2, 3, 3, 5, 7], [0, 1, 2, 3, 4, 7], [0, 1, 2, 3, 5, 7], [0, 6, 7]]
>>>print test.allPaths([[0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
[0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.8, 0.0, 0.0, 0.0, .05, .15, 0.0, 0.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0],
[0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
[0.9, 0.0, 0.1, 0.0, 0.0, 0.0, 0.0, 0.0]],0,7)
错误似乎发生在“flag = flag * len(adjMat)”行。有什么建议吗?
答案 0 :(得分:1)
每次递归调用都会将flag
列表的大小增加len(adjMat)
。
对函数的第一次调用使用带有len(adjMat)
元素的标志列表,并将其传递给递归调用。在那里,列表将乘以len(adjMat)
,从而生成len(adjMat) * len(adjMat)
个元素。通过一些递归调用,这很快就会失控,你可能会耗尽内存来存储这个过大的flag
列表。