问题描述
我一直在使用递归+记忆法做0-1背包问题。
我的代码:
def knapSack(W,wt,val,n):
'''
:param W: capacity of knapsack
:param wt: list containing weights
:param val: list containing corresponding values
:param n: size of lists
:return: Integer
'''
t = [[-1 for x in range(W + 1)] for j in range(n + 1)]
if n == 0 or W == 0:
return 0
if t[n][W] != -1:
return t[n][W]
elif wt[n-1] <= W:
t[n][W] = max(val[n-1]+knapSack(W-wt[n-1],n-1),knapSack(W,n-1))
return t[n][W]
elif wt[n-1] > W:
t[n][W] = knapSack(wt,W,n-1)
return t[n][W]
为什么会出现运行时错误。
Runtime ErrorTraceback (most recent call last):
File "/home/e8c2fc67721232cbee976a6adfc2c990.py",line 41,in <module>
print(knapSack(W,n))
File "/home/e8c2fc67721232cbee976a6adfc2c990.py",line 12,in knapSack
t[n][W] = knapSack(wt,n-1)
File "/home/e8c2fc67721232cbee976a6adfc2c990.py",line 5,in knapSack
t = [[-1 for x in range(W + 1)] for j in range(n + 1)]
File "/home/e8c2fc67721232cbee976a6adfc2c990.py",in <listcomp>
t = [[-1 for x in range(W + 1)] for j in range(n + 1)]
TypeError: can only concatenate list (not "int") to list
这是我遇到的运行时错误。
解决方法
我怀疑您的函数定义的参数顺序与参数顺序之间存在不匹配,在该参数中args实际传递给以下其中一个许多调用。
请注意,在追溯中:
在第41行,其arg顺序显示为print(knapSack(W,wt,val,n))
第12行的行显示了不同的arg顺序knapSack(wt,W,n-1)
对于条件elif wt[n-1] > W:
,您对knapSack()
的递归调用肯定会以错误的顺序传递args。
如果您将循环放在第一个if下面,我认为可以轻松解决此问题。
if n == 0 or W == 0:
return 0
t = [[-1 for x in range(W + 1)] for j in range(n + 1)]
if t[n][W] != -1:
return t[n][W]
elif wt[n-1] <= W:
t[n][W] = max(val[n-1]+knapSack(W-wt[n-1],n-1),knapSack(W,n-1))
return t[n][W]
elif wt[n-1] > W:
t[n][W] = knapSack(wt,n-1)
return t[n][W]