动态规划:《完全背包问题》-python实现

接上一篇博客

在01背包问题中,每个物体都不允许重复使用。而完全背包问题:有n种重量和价值分别为Wi和Vi的物品。从这些物品中挑选出总重量不超过w的物品,每种物品都可以挑选多件,求所有挑选方案中价值总和的最大值。

#coding:utf-8
#完全背包问题
import sys


def track(d, c, w):
    x = []
    while c > 0:                                                #因为可以重复,所以每次都需要重新寻找w
        for i in range(len(w), 1, -1):
            if d[i][c] != d[i - 1][c]:
                x.append(w[i - 1])
                c = c - w[i - 1]
                break

    if d[1][c] > 0:
        x.append(w[0])
    return x

if __name__ == '__main__':
    c = input()                                                 #输入一个限制条件,例如背包的体积为c
    w = sys.stdin.readline().strip().split(' ')                 #每个物品的体积
    w = map(int, w)
    v = sys.stdin.readline().strip().split(' ')                 #对应每个物体的价值
    v = map(int, v)

    dp = [[0] * (c + 1) for i in range(len(w) + 1)]

    for i in range(1, len(w) + 1):
        for j in range(1, c + 1):
            if j >= w[i - 1]:
                dp[i][j] = max(dp[i - 1][j], dp[i][j - w[i - 1]] + v[i - 1])
            else:
                dp[i][j] = dp[i - 1][j]

    print max(dp[len(w)])

    print track(dp, c, w)

猜你喜欢

转载自blog.csdn.net/w113691/article/details/81749364
今日推荐