动态规划：《部分背包问题》-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)