NC145:0-1背包问题(python)
这段代码不仅可以求出背包可以装多重的东西,还可以求出背包装了什么。
已知一个背包最多能容纳体积之和为v的物品
现有 n 个物品,第 i 个物品的体积为 vi , 重量为 wi
求当前背包最多能装多大重量的物品?
n=4
V=8
v=[2,3,4,5]#每个物品的体积
w=[3,4,5,8]#每个物品的重量
l=[[0 for i in range(V+1)] for j in range(n+1)]
#t用于存放当前背包下要装的物品的'id';t[i][j]用于存放容积为j,能放入i个物品的背包的物品
t=[[[] for i in range(V+1)] for j in range(n+1)]
for i in range(1,n+1):
for j in range(1,V+1):
if v[i-1]>j or l[i-1][j]>=l[i-1][j-v[i-1]]+w[i-1]:
l[i][j]=l[i-1][j]
t[i][j]=t[i-1][j]
else: #只需要在装入新的物品的时候改变t矩阵
l[i][j]=l[i-1][j-v[i-1]]+w[i-1]
t[i][j].append(i)
if len(t[i-1][j-v[i-1]])>0:
#如果t[i-1][j-v[i-1]] 为[],则说明背包没装东西
for k in range(len(t[i-1][j-v[i-1]])): #背包中可能装了很多东西
t[i][j].append(t[i-1][j-v[i-1]][-1])
print(l[n][V])
print(t[-1][-1])```