在将什么是0-1背包问题前,先来看下面一道例题:
题意概要:有 n 个物品和一个容量为 W 的背包,每个物品有重量w i和价值v i两种属性,要求 选若干物品放入背包使背包中物品的总价值最大且背包中物品的总重量不超过背包的容量。
在上述例题中,由于每个物体只有两种可能的状态(取与不取),对应二进制中 的 0 和 1,这类问题便被称为「0-1 背包问题」。
解释
例题中已知条件有第 i 个物品的重量 wi,价值 vi,以及背包的总容量 W。
设 DP 状态 fi,j 为在只能放前 i 个物品的情况下,容量为 j 的背包所能达到的最大 总价值。
考虑转移。假设当前已经处理好了前 i − 1 个物品的所有状态,那么对于第 i 个物 品,当其不放入背包时,背包的剩余容量不变,背包中物品的总价值也不变,故 这种情况的最大价值为 fi−1,j;当其放入背包时,背包的剩余容量会减小 wi,背包 中物品的总价值会增大 vi,故这种情况的最大价值为 fi−1,j−wi + vi。
由此可以得出状态转移方程:
这里如果直接采用二维数组对状态进行记录,会出现 MLE。可以考虑改用滚动数 组的形式来优化。
由于对 fi 有影响的只有 fi−1,可以去掉第一维,直接用 fi 来表示处理到当前物品 时背包容量为 i 的最大价值,得出以下方程:
务必牢记并理解这个转移方程,因为大部分背包问题的转移方程都是在此基础上推导出来的。
实现
还有一点需要注意的是,很容易写出这样的 错误核心代码:
// C++ Version
for (int i = 1; i <= n; i++)
for (int l = 0; l <= W - w[i]; l++)
f[l + w[i]] = max(f[l] + v[i], f[l + w[i]]);
// 由 f[i][l + w[i]] = max(max(f[i - 1][l + w[i]],f[i - 1][l] + w[i]),f[i][l +
// w[i]]); 简化而来
这段代码哪里错了呢?枚举顺序错了。
仔细观察代码可以发现:对于当前处理的物品 i 和当前状态 fi,j,在 j ⩾ wi 时, fi,j 是会被 fi,j−wi 所影响的。这就相当于物品 i 可以多次被放入背包,与题意不 符。(事实上,这正是完全背包问题的解法)
为了避免这种情况发生,我们可以改变枚举的顺序,从 W 枚举到 wi,这样就不 会出现上述的错误,因为 fi,j 总是在 fi,j−wi 前被更新。
// C++ Version
for (int i = 1; i <= n; i++)
for (int l = W; l >= w[i]; l--) f[l] = max(f[l], f[l - w[i]] + v[i]);
例题完整代码
#include <iostream>
using namespace std;
const int maxn = 13010;
int n, W, w[maxn], v[maxn], f[maxn];
int main() {
cin >> n >> W;
for (int i = 1; i <= n; i++) cin >> w[i] >> v[i]; // 读入数据
for (int i = 1; i <= n; i++)
for (int l = W; l >= w[i]; l--)
if (f[l - w[i]] + v[i] > f[l]) f[l] = f[l - w[i]] + v[i]; // 状态转移方程
cout << f[W];
return 0;
}
下一篇,将会讲完全背包问题。