【ACWing】5. 多重背包问题 II

题目地址：

https://www.acwing.com/problem/content/5/

有$N$种物品和一个容量是$V$的背包。第$i$种物品最多有$s_i$件，每件体积是$v_i$，价值是$w_i$。求解将哪些物品装入背包，可使物品体积总和不超过背包容量，且价值总和最大。输出最大价值。

输入格式：
第一行两个整数，$N$，$V$，用空格隔开，分别表示物品种数和背包容积。接下来有$N$行，每行三个整数$v_i,w_i,s_i$，用空格隔开，分别表示第$i$种物品的体积、价值和数量。

输出格式：
输出一个整数，表示最大价值。

数据范围：
$0<N\le 1000$
$0<V\le 2000$
$0<v_i,w_i,s_i\le 2000$

思路是动态规划。但这里需要用二进制优化，其基本思路是，将每个物品的个数$s$拆成若干个形如$2^k$的个数的”打包”，这样就将多重背包问题转为了$0-1$背包问题了，从而将时间复杂度降为$O(V{\sum }_{i=1}^N\log s_i)$。例如，如果一个物品有$10$个，那么可以将它拆成$1+2+2^2+3$，这样这个物品的所有取法，一定包含于将其拆成若干小块，每个小块最多取一个的取法。

分解的存在性证明如下：设$s=1+2+2^2+...+2^k+c,c<2^{k+1}$，那么 { 0 , 1 , 2 , . . . , 2 k + 1 − 1 } \{0,1,2,...,2^{k+1}-1\} {0,1,2,...,2k+1−1}里的整数显然能被 { 1 , 2 , 2 2 , . . . , 2 k } \{1,2,2^2,...,2^{k}\} {1,2,22,...,2k}的组合表示出来，只需要做二进制表示，然后分解出来即可。那么将这个集合里的每个数都加上$c$，就得到了 { c , c + 1 , . . . , c + 2 k + 1 − 1 = s } \{c,c+1,...,c+2^{k+1}-1=s\} {c,c+1,...,c+2k+1−1=s}这些数。由于$c\le 2^{k+1}-1$，所以 { 0 , 1 , . . . , s } \{0,1,...,s\} {0,1,...,s}都可以被表示出来。

代码如下：

#include <iostream>
using namespace std;

const int N = 15000;
int n, m;
int v[N], w[N], cnt;
int f[N];

int main() {
  scanf("%d%d", &n, &m);
  for (int i = 1; i <= n; i++) {
    int a, b, s;
    scanf("%d%d%d", &a, &b, &s);
    for (int k = 1; k <= s; k <<= 1) {
      v[++cnt] = a * k;
      w[cnt] = b * k;
      s -= k;
    }

    if (s) {
      v[++cnt] = a * s;
      w[cnt] = b * s;
    }
  }

  n = cnt;
  for (int i = 1; i <= n; i++)
    for (int j = m; j >= v[i]; j--) 
      f[j] = max(f[j], f[j - v[i]] + w[i]);

  printf("%d\n", f[m]);
}

时间复杂度$O(V{\sum }_{i=1}^N\log s_i)$，空间$O({\sum }_{i=1}^N\log s_i-N)$。代码里$N$开$15000$是$1000\times {\log }_{2}2000$算出来的。