C++0-1背包问题
背包问题是一个非常好的动态规划入门题目,是动态规划中最常见的问题形式。其具有很多变种,例如 0-1 背包问题、完全背包问题、多重背包问题等。
在本节中,我们先来求解最常见的 0-1 背包问题。
Question
给定
观察图 14-17 ,由于物品编号
图 14-17 0-1 背包的示例数据
我们可以将 0-1 背包问题看作是一个由
该问题的目标是求解“在限定背包容量下的最大价值”,因此较大概率是个动态规划问题。
第一步:思考每轮的决策,定义状态,从而得到
对于每个物品来说,不放入背包,背包容量不变;放入背包,背包容量减小。由此可得状态定义:当前物品编号
状态
待求解的是
第二步:找出最优子结构,进而推导出状态转移方程
当我们做出物品
- 不放入物品
:背包容量不变,状态变化为 。 - 放入物品
:背包容量减小 ,价值增加 ,状态变化为 。
上述分析向我们揭示了本题的最优子结构:最大价值
需要注意的是,若当前物品重量
第三步:确定边界条件和状态转移顺序
当无物品或无剩余背包容量时最大价值为
当前状态
根据以上分析,我们接下来按顺序实现暴力搜索、记忆化搜索、动态规划解法。
1. 方法一:暴力搜索
搜索代码包含以下要素。
- 递归参数:状态
。 - 返回值:子问题的解
。 - 终止条件:当物品编号越界
或背包剩余容量为 时,终止递归并返回价值 。 - 剪枝:若当前物品重量超出背包剩余容量,则只能不放入背包。
knapsack.cpp
/* 0-1 背包:暴力搜索 */
int knapsackDFS(vector<int> &wgt, vector<int> &val, int i, int c) {
// 若已选完所有物品或背包无容量,则返回价值 0
if (i == 0 || c == 0) {
return 0;
}
// 若超过背包容量,则只能不放入背包
if (wgt[i - 1] > c) {
return knapsackDFS(wgt, val, i - 1, c);
}
// 计算不放入和放入物品 i 的最大价值
int no = knapsackDFS(wgt, val, i - 1, c);
int yes = knapsackDFS(wgt, val, i - 1, c - wgt[i - 1]) + val[i - 1];
// 返回两种方案中价值更大的那一个
return max(no, yes);
}
如图 14-18 所示,由于每个物品都会产生不选和选两条搜索分支,因此时间复杂度为
观察递归树,容易发现其中存在重叠子问题,例如
图 14-18 0-1 背包的暴力搜索递归树
2. 方法二:记忆化搜索
为了保证重叠子问题只被计算一次,我们借助记忆列表 mem
来记录子问题的解,其中 mem[i][c]
对应
引入记忆化之后,时间复杂度取决于子问题数量,也就是
knapsack.cpp
/* 0-1 背包:记忆化搜索 */
int knapsackDFSMem(vector<int> &wgt, vector<int> &val, vector<vector<int>> &mem, int i, int c) {
// 若已选完所有物品或背包无容量,则返回价值 0
if (i == 0 || c == 0) {
return 0;
}
// 若已有记录,则直接返回
if (mem[i][c] != -1) {
return mem[i][c];
}
// 若超过背包容量,则只能不放入背包
if (wgt[i - 1] > c) {
return knapsackDFSMem(wgt, val, mem, i - 1, c);
}
// 计算不放入和放入物品 i 的最大价值
int no = knapsackDFSMem(wgt, val, mem, i - 1, c);
int yes = knapsackDFSMem(wgt, val, mem, i - 1, c - wgt[i - 1]) + val[i - 1];
// 记录并返回两种方案中价值更大的那一个
mem[i][c] = max(no, yes);
return mem[i][c];
}
图 14-19 展示了在记忆化递归中被剪掉的搜索分支。
图 14-19 0-1 背包的记忆化搜索递归树
3. 方法三:动态规划¶
动态规划实质上就是在状态转移中填充
knapsack.cpp
/* 0-1 背包:动态规划 */
int knapsackDP(vector<int> &wgt, vector<int> &val, int cap) {
int n = wgt.size();
// 初始化 dp 表
vector<vector<int>> dp(n + 1, vector<int>(cap + 1, 0));
// 状态转移
for (int i = 1; i <= n; i++) {
for (int c = 1; c <= cap; c++) {
if (wgt[i - 1] > c) {
// 若超过背包容量,则不选物品 i
dp[i][c] = dp[i - 1][c];
} else {
// 不选和选物品 i 这两种方案的较大值
dp[i][c] = max(dp[i - 1][c], dp[i - 1][c - wgt[i - 1]] + val[i - 1]);
}
}
}
return dp[n][cap];
}
如图 14-20 所示,时间复杂度和空间复杂度都由数组 dp
大小决定,即
图 14-20 0-1 背包的动态规划过程
4. 空间优化
由于每个状态都只与其上一行的状态有关,因此我们可以使用两个数组滚动前进,将空间复杂度从
进一步思考,我们是否可以仅用一个数组实现空间优化呢?观察可知,每个状态都是由正上方或左上方的格子转移过来的。假设只有一个数组,当开始遍历第
- 如果采取正序遍历,那么遍历到
时,左上方 ~ 值可能已经被覆盖,此时就无法得到正确的状态转移结果。 - 如果采取倒序遍历,则不会发生覆盖问题,状态转移可以正确进行。
图 14-21 展示了在单个数组下从第
图 14-21 0-1 背包的空间优化后的动态规划过程
在代码实现中,我们仅需将数组 dp
的第一维
knapsack.cpp
/* 0-1 背包:空间优化后的动态规划 */
int knapsackDPComp(vector<int> &wgt, vector<int> &val, int cap) {
int n = wgt.size();
// 初始化 dp 表
vector<int> dp(cap + 1, 0);
// 状态转移
for (int i = 1; i <= n; i++) {
// 倒序遍历
for (int c = cap; c >= 1; c--) {
if (wgt[i - 1] <= c) {
// 不选和选物品 i 这两种方案的较大值
dp[c] = max(dp[c], dp[c - wgt[i - 1]] + val[i - 1]);
}
}
}
return dp[cap];
}