背包问题:多重背包问题II

319 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

往期

  1. 01背包问题
  2. 完全背包问题
  3. 多重背包问题I

题目

多重背包问题II

NN 种物品和一个容量是 VV 的背包。

ii 种物品最多有 sis_{i} 件,每件体积是 viv_{i},价值是 wiw_{i}

求解将哪些物品装入背包,可使物品体积总和不超过背包容量,且价值总和最大。 输出最大价值。

输入格式

第一行两个整数,NNVV,用空格隔开,分别表示物品种数和背包容积。

接下来有 NN 行,每行三个整数 viv_{i},wiw_{i},sis_{i},用空格隔开,分别表示第 ii 种物品的体积、价值和数量。

输出格式

输出一个整数,表示最大价值。

数据范围

0<N10000< N\le 1000

0<V20000< V\le 2000

0<vi,wi,si20000< v_{i},w_{i},s_{i}\le 2000

输入样例

4 5
1 2 3
2 4 1
3 4 3
4 5 2

输出样例

10

解题思路

相比多重背包问题I,本题数据范围变大,需优化时间复杂度

不能用完全背包问题抛弃k循环的思路来优化这个问题,因为每个物品的件数不同,不能像完全背包问题那样优化状态转移方程

多重背包的二进制优化

考虑将多重背包问题转化为01背包问题,考虑把第ii种物品换成若干件物品,使得原问题中第ii种物品可取的每种策略(取00件、取11件、...、取kk件),均能等价于取若干件转换以后的物品,取超过kk件的策略必不能出现。

若直接把第ii件物品换成s[i]s[i]件01背包中的物品,时间复杂度不变。

考虑二进制的思想,将kk件物品拆成log2k\log_{2}{k}件新的物品,k=1+2+22+...+2t+c(c2t+1)k=1+2+2^{2}+...+2^{t}+c\left ( c\le 2^{t+1} \right ),用这log2k\log_{2}{k}个新数,可以凑出0k0 \sim k中的任何一个数。这log2k\log_{2}{k}个数作为新物品的系数,新物品的体积和价值就是原物品的费用和价值乘以这个系数。

eg: 13=1+2+4+613=1+2+4+6,最多取13件的物品被分成系数分别为1,2,4,61,2,4,6的四件物品,原先要枚举14次,拆分之后只需枚举5次;这种优化对于大数尤其明显,例如最多取1024件的物品,在正常情况下要枚举1025次 , 二进制思想下转化成01背包只需要枚举11次。

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        
        int n = sc.nextInt();               // 物品种数
        int m = sc.nextInt();               // 背包容积
        
        int[] v = new int[n];
        int[] w = new int[n];
        int[] s = new int[n];
        
        
        for (int i = 0; i < n; i++) {
            v[i] = sc.nextInt();
            w[i] = sc.nextInt();
            s[i] = sc.nextInt();
        }
        
        // 二进制拆分
        int[] newV = new int[n*11];    // 2^11 = 2048,也就是说s最多可以拆成11个,故数组容量乘以11
        int[] newW = new int[n*11];
        int newN = 0;                   // 新的物品种数
        for (int i = 0; i < n; i++) {
            for (int j = 1; j <= s[i]; j *= 2) {
                newV[newN] = j * v[i];   // 体积
                newW[newN] = j * w[i];   // 价值
                s[i] -= j;
                newN++;
            }
            if (s[i] > 0) {
                newV[newN] = s[i] * v[i];
                newW[newN] = s[i] * w[i];
                newN++;
            }
        }
        
        // 01背包问题
        int[] dp = new int[m+1];        
        for (int i = 0; i < newN; i++) {
            for (int j = m; j >= newV[i]; j--) {
                dp[j] = Math.max(dp[j], dp[j-newV[i]]+newW[i]);
            }
        }
        System.out.println(dp[m]);
    }
}

Reference

  1. 背包问题九讲
  2. 多重背包的二进制拆分代码
  3. 多重背包问题的优化思路
  4. 多重背包问题的优化思路