背包问题理论基础多重背包 - 图1
参与本项目,贡献其他语言版本的代码,拥抱开源,让更多学习算法的小伙伴们收益!

动态规划:关于多重背包,你该了解这些!

之前我们已经系统的讲解了01背包和完全背包,如果没有看过的录友,建议先把如下三篇文章仔细阅读一波。

这次我们再来说一说多重背包

多重背包

对于多重背包,我在力扣上还没发现对应的题目,所以这里就做一下简单介绍,大家大概了解一下。

有N种物品和一个容量为V 的背包。第i种物品最多有Mi件可用,每件耗费的空间是Ci ,价值是Wi 。求解将哪些物品装入背包可使这些物品的耗费的空间 总和不超过背包容量,且价值总和最大。

多重背包和01背包是非常像的, 为什么和01背包像呢?

每件物品最多有Mi件可用,把Mi件摊开,其实就是一个01背包问题了。

例如:

背包最大重量为10。

物品为:

重量 价值 数量
物品0 1 15 2
物品1 3 20 3
物品2 4 30 2

问背包能背的物品最大价值是多少?

和如下情况有区别么?

重量 价值 数量
物品0 1 15 1
物品0 1 15 1
物品1 3 20 1
物品1 3 20 1
物品1 3 20 1
物品2 4 30 1
物品2 4 30 1

毫无区别,这就转成了一个01背包问题了,且每个物品只用一次。

这种方式来实现多重背包的代码如下:

  1. void test_multi_pack() {
  2. vector<int> weight = {1, 3, 4};
  3. vector<int> value = {15, 20, 30};
  4. vector<int> nums = {2, 3, 2};
  5. int bagWeight = 10;
  6. for (int i = 0; i < nums.size(); i++) {
  7. while (nums[i] > 1) { // nums[i]保留到1,把其他物品都展开
  8. weight.push_back(weight[i]);
  9. value.push_back(value[i]);
  10. nums[i]--;
  11. }
  12. }
  13. vector<int> dp(bagWeight + 1, 0);
  14. for(int i = 0; i < weight.size(); i++) { // 遍历物品
  15. for(int j = bagWeight; j >= weight[i]; j--) { // 遍历背包容量
  16. dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
  17. }
  18. for (int j = 0; j <= bagWeight; j++) {
  19. cout << dp[j] << " ";
  20. }
  21. cout << endl;
  22. }
  23. cout << dp[bagWeight] << endl;
  24. }
  25. int main() {
  26. test_multi_pack();
  27. }
  • 时间复杂度:O(m × n × k),m:物品种类个数,n背包容量,k单类物品数量

也有另一种实现方式,就是把每种商品遍历的个数放在01背包里面在遍历一遍。

代码如下:(详看注释)

  1. void test_multi_pack() {
  2. vector<int> weight = {1, 3, 4};
  3. vector<int> value = {15, 20, 30};
  4. vector<int> nums = {2, 3, 2};
  5. int bagWeight = 10;
  6. vector<int> dp(bagWeight + 1, 0);
  7. for(int i = 0; i < weight.size(); i++) { // 遍历物品
  8. for(int j = bagWeight; j >= weight[i]; j--) { // 遍历背包容量
  9. // 以上为01背包,然后加一个遍历个数
  10. for (int k = 1; k <= nums[i] && (j - k * weight[i]) >= 0; k++) { // 遍历个数
  11. dp[j] = max(dp[j], dp[j - k * weight[i]] + k * value[i]);
  12. }
  13. }
  14. // 打印一下dp数组
  15. for (int j = 0; j <= bagWeight; j++) {
  16. cout << dp[j] << " ";
  17. }
  18. cout << endl;
  19. }
  20. cout << dp[bagWeight] << endl;
  21. }
  22. int main() {
  23. test_multi_pack();
  24. }
  • 时间复杂度:O(m × n × k),m:物品种类个数,n背包容量,k单类物品数量

从代码里可以看出是01背包里面在加一个for循环遍历一个每种商品的数量。 和01背包还是如出一辙的。

当然还有那种二进制优化的方法,其实就是把每种物品的数量,打包成一个个独立的包。

和以上在循环遍历上有所不同,因为是分拆为各个包最后可以组成一个完整背包,具体原理我就不做过多解释了,大家了解一下就行,面试的话基本不会考完这个深度了,感兴趣可以自己深入研究一波。

总结

多重背包在面试中基本不会出现,力扣上也没有对应的题目,大家对多重背包的掌握程度知道它是一种01背包,并能在01背包的基础上写出对应代码就可以了。

至于背包九讲里面还有混合背包,二维费用背包,分组背包等等这些,大家感兴趣可以自己去学习学习,这里也不做介绍了,面试也不会考。

其他语言版本

Java:

  1. public void testMultiPack1(){
  2. // 版本一:改变物品数量为01背包格式
  3. List<Integer> weight = new ArrayList<>(Arrays.asList(1, 3, 4));
  4. List<Integer> value = new ArrayList<>(Arrays.asList(15, 20, 30));
  5. List<Integer> nums = new ArrayList<>(Arrays.asList(2, 3, 2));
  6. int bagWeight = 10;
  7. for (int i = 0; i < nums.size(); i++) {
  8. while (nums.get(i) > 1) { // 把物品展开为i
  9. weight.add(weight.get(i));
  10. value.add(value.get(i));
  11. nums.set(i, nums.get(i) - 1);
  12. }
  13. }
  14. int[] dp = new int[bagWeight + 1];
  15. for(int i = 0; i < weight.size(); i++) { // 遍历物品
  16. for(int j = bagWeight; j >= weight.get(i); j--) { // 遍历背包容量
  17. dp[j] = Math.max(dp[j], dp[j - weight.get(i)] + value.get(i));
  18. }
  19. System.out.println(Arrays.toString(dp));
  20. }
  21. }
  22. public void testMultiPack2(){
  23. // 版本二:改变遍历个数
  24. int[] weight = new int[] {1, 3, 4};
  25. int[] value = new int[] {15, 20, 30};
  26. int[] nums = new int[] {2, 3, 2};
  27. int bagWeight = 10;
  28. int[] dp = new int[bagWeight + 1];
  29. for(int i = 0; i < weight.length; i++) { // 遍历物品
  30. for(int j = bagWeight; j >= weight[i]; j--) { // 遍历背包容量
  31. // 以上为01背包,然后加一个遍历个数
  32. for (int k = 1; k <= nums[i] && (j - k * weight[i]) >= 0; k++) { // 遍历个数
  33. dp[j] = Math.max(dp[j], dp[j - k * weight[i]] + k * value[i]);
  34. }
  35. System.out.println(Arrays.toString(dp));
  36. }
  37. }
  38. }

Python:

  1. def test_multi_pack1():
  2. '''版本一:改变物品数量为01背包格式'''
  3. weight = [1, 3, 4]
  4. value = [15, 20, 30]
  5. nums = [2, 3, 2]
  6. bag_weight = 10
  7. for i in range(len(nums)):
  8. # 将物品展开数量为1
  9. while nums[i] > 1:
  10. weight.append(weight[i])
  11. value.append(value[i])
  12. nums[i] -= 1
  13. dp = [0]*(bag_weight + 1)
  14. # 遍历物品
  15. for i in range(len(weight)):
  16. # 遍历背包
  17. for j in range(bag_weight, weight[i] - 1, -1):
  18. dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
  19. print(" ".join(map(str, dp)))
  20. def test_multi_pack2():
  21. '''版本:改变遍历个数'''
  22. weight = [1, 3, 4]
  23. value = [15, 20, 30]
  24. nums = [2, 3, 2]
  25. bag_weight = 10
  26. dp = [0]*(bag_weight + 1)
  27. for i in range(len(weight)):
  28. for j in range(bag_weight, weight[i] - 1, -1):
  29. # 以上是01背包,加上遍历个数
  30. for k in range(1, nums[i] + 1):
  31. if j - k*weight[i] >= 0:
  32. dp[j] = max(dp[j], dp[j - k*weight[i]] + k*value[i])
  33. print(" ".join(map(str, dp)))
  34. if __name__ == '__main__':
  35. test_multi_pack1()
  36. test_multi_pack2()

Go:

  1. package theory
  2. import "log"
  3. // 多重背包可以化解为 01 背包
  4. func multiplePack(weight, value, nums []int, bagWeight int) int {
  5. for i := 0; i < len(nums); i++ {
  6. for nums[i] > 1 {
  7. weight = append(weight, weight[i])
  8. value = append(value, value[i])
  9. nums[i]--
  10. }
  11. }
  12. log.Println(weight)
  13. log.Println(value)
  14. res := make([]int, bagWeight+1)
  15. for i := 0; i < len(weight); i++ {
  16. for j := bagWeight; j >= weight[i]; j-- {
  17. res[j] = getMax(res[j], res[j-weight[i]]+value[i])
  18. }
  19. log.Println(res)
  20. }
  21. return res[bagWeight]
  22. }

单元测试

  1. package theory
  2. import "testing"
  3. func Test_multiplePack(t *testing.T) {
  4. type args struct {
  5. weight []int
  6. value []int
  7. nums []int
  8. bagWeight int
  9. }
  10. tests := []struct {
  11. name string
  12. args args
  13. want int
  14. }{
  15. {
  16. name: "one",
  17. args: args{
  18. weight: []int{1, 3, 4},
  19. value: []int{15, 20, 30},
  20. nums: []int{2, 3, 2},
  21. bagWeight: 10,
  22. },
  23. want: 90,
  24. },
  25. }
  26. for _, tt := range tests {
  27. t.Run(tt.name, func(t *testing.T) {
  28. if got := multiplePack(tt.args.weight, tt.args.value, tt.args.nums, tt.args.bagWeight); got != tt.want {
  29. t.Errorf("multiplePack() = %v, want %v", got, tt.want)
  30. }
  31. })
  32. }
  33. }

输出

  1. === RUN Test_multiplePack
  2. === RUN Test_multiplePack/one
  3. 2022/03/02 21:09:05 [1 3 4 1 3 3 4]
  4. 2022/03/02 21:09:05 [15 20 30 15 20 20 30]
  5. 2022/03/02 21:09:05 [0 15 15 15 15 15 15 15 15 15 15]
  6. 2022/03/02 21:09:05 [0 15 15 20 35 35 35 35 35 35 35]
  7. 2022/03/02 21:09:05 [0 15 15 20 35 45 45 50 65 65 65]
  8. 2022/03/02 21:09:05 [0 15 30 30 35 50 60 60 65 80 80]
  9. 2022/03/02 21:09:05 [0 15 30 30 35 50 60 60 70 80 80]
  10. 2022/03/02 21:09:05 [0 15 30 30 35 50 60 60 70 80 80]
  11. 2022/03/02 21:09:05 [0 15 30 30 35 50 60 60 70 80 90]
  12. --- PASS: Test_multiplePack (0.00s)
  13. --- PASS: Test_multiplePack/one (0.00s)
  14. PASS

背包问题理论基础多重背包 - 图2