python背包算法_动态规划：《部分背包问题》-python实现

接上两篇

部分背包问题:有n种重量和价值分别为Wi和Vi的物品。从这些物品中挑选出总重量不超过w的物品，第i种物品最多选mi个，求所有挑选方案中价值总和的最大值。

将部分背包问题转换成01背包问题进行求解。

import sys

def track(d, c, w):

x = []

for i in range(1, len(w)):

if d[i][c] != d[i - 1][c]:

x.append(w[i - 1])

c = c - w[i - 1]

if d[1][c] > 0:

x.append(w[0])

return x

if __name__ == '__main__':

n = input()

w = sys.stdin.readline().strip().split(' ')

w = map(int, w)

v = sys.stdin.readline().strip().split(' ')

v = map(int, v)

m = sys.stdin.readline().strip().split(' ')

m = map(int, m)

dp = [[0] * (n + 1) for i in range(sum(m) + 1)]

ww = []

vv = []

for i in range(0, len(w)):

for j in range(1, m[i] + 1):

ww.append(w[i] * j)

vv.append(v[i] * j)

print ww

print vv

for i in range(1, len(ww) + 1):

for j in range(1, n + 1):

if j >= ww[i - 1]:

dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - ww[i - 1]] + vv[i - 1])

else:

dp[i][j] = dp[i - 1][j]

print max(dp[len(ww)])

print track(dp, n, ww)