变种 背包问题_AcWing 4. 多重背包问题 I---01背包问题的变种

1. 基本解题思路：

一个基本思路是，将此问题转换为01背包求解！

比如物品1有3件，每件价值为2，我们不妨创建3个物品1，存在数组v和数组w中

最终更新一下总物品数n即可，然后套用01背包问题进行求解。

Python3 代码

class Solution:

# 这里的代码和01背包问题代码一样，只是在输入时，对数组v,w和物品件数n进行了调整

def max_value(self, n, m, v, w):

dp = [0] * (m + 1)

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

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

dp[j] = max(dp[j], dp[j - v[i]] + w[i])

return dp[m]

if __name__ == '__main__':

import sys

n, m = map(int, input().split())

lines = sys.stdin.readlines()

v = [0]

w = [0]

n = 0

for line in lines:

line = list(map(int, line.split()))

v.extend([line[0]] * line[2])

w.extend([line[1]] * line[2])

n += line[2]

print(Solution().max_value(n, m, v, w))

2. 优化时间复杂度到O(mlogn) (多重背包问题II的解法)

前面提到的多重背包问题的解法，是把多件物品，转换成多个单件物品，添加到v和w数组中。

如10件物品A, 则插入十条记录到v和w数组中，实际上这一过程可以进行优化！

我们可以把十件物品A分成若干份，这若干份必须可以组合成0～10以内的任何一个数字。

做法是：1,2,4,...,2^(k-1),10-2^k+1

即：10可以分为 1，2，4，3

显然这四个数字，可以组合成0～10以内的任何一个数字，如 8 = 1 + 4 + 3

每一份对应的体积和价值，用系数乘以1件物品的体积和价值。

这么做的好处，可以把时间复杂度从O(nm)降为O(m log n)，剩下的继续用01背包问题的解法求解。

Python3 代码

class Solution:

def max_value(self, n, m, v, w):

dp = [0] * (m + 1)

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

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

dp[j] = max(dp[j], dp[j - v[i]] + w[i])

return dp[-1]

if __name__ == '__main__':

import sys

n, m = map(int, input().split())

lines = sys.stdin.readlines()

v, w = [0], [0]

n = 0

for line in lines:

line = list(map(int, line.split()))

k = 1

while k <= line[2]: # 假设line[2]=13,k取1,2,4之后，line[2] = 6 < k = 8 退出循环

v.append(k * line[0])

w.append(k * line[1])

line[2] -= k

k *= 2

n += 1 # 物品总数加1

if line[2]:

v.append(line[2] * line[0])

w.append(line[2] * line[1])

n += 1

print(Solution().max_value(n, m, v, w))