一、问题描述
有 N 种物品,第 i 种物品的体积是 ci,价值是 wi,每种物品的数量都是有限的,为 ni。 现有容量为 V 的背包,请你放入若干物品,在总体积不超过 V 的条件下,使总价值尽可能大。
二、解法
转化为01背包:
for (int i = 1; i <= N; i++) {
for (int j = 0; j <= V; j++) {
for (int k = 0; k <= n[i]; k++) {
if (j >= c[i] * k) {
dp[i][j] = max(dp[i - 1][j - c[i] * k] + w[i] * k, dp[i][j]);
}
}
}
}
这一份代码和 01 背包相比,不再有else部分了,因为,k = 0的时候dp[i][j] = max(dp[i - 1][j], dp[i][j]),相当于 01 背包的else部分。
空间优化:
既然多重背包的可以转换成 01 背包,那么我们必然也可以像 01 背包那样优化空间复杂度。还是按照从大到小的顺序枚举背包体积。
for (int i = 1; i <= N; i++) {
for (int j = V; j >= 0; j--) {
for (int k = 1; k <= n[i]; k++) {
if (j >= c[i] * k) {
dp[j] = max(dp[j - c[i] * k] + w[i] * k, dp[j]);
}
}
}
}
三、二进制优化
证明:
于是,n 件物品,就拆分成至多 logn 件物品。 再转化为 01 背包问题求解,原问题复杂度降低到 O(V∑logn)。 示例代码
#include <cstdio>
#include <cstring>
#define max(a, b) ((a) > (b) ? (a) : (b))
using namespace std;
const int N = 25000, M = 2005;;
int n, V, v[N], w[N];
int dp[M];
void f() {
//将多重背包问题转化为 01 背包
for (int i = 1; i <= n; i++) {
for (int j = V; j >= v[i]; j--){
dp[j] = max(dp[j], dp[j - v[i]] + w[i]);
}
}
}
int main() {
scanf("%d%d", &n, &V);
int cnt = 0, a, b, s;
for (int i = 1; i <= n; i++) {
scanf("%d%d%d", &a, &b, &s);
//分解为log(s)个物品
int k = 1;
while (k <= s) {
s-= k;
v[++cnt] = k * a;
w[cnt] = k * b;
k <<= 1;
}
if (s > 0) {
v[++cnt] = s * a;
w[cnt] = s * b;
}
}
n = cnt;
f();
printf("%d\n", dp[V]);
return 0;
}