背包问题
动态规划基本步骤
1.确定dp数组的含义
2.确定递推公式
3.dp数组初始化
4.确定遍历顺序
5.代码优化
01背包
问题描述:有N件物品和⼀个最多能被重量为W 的背包。第i件物品的重量是weight[i],得到的价值是value[i] 。每件物品只能⽤⼀次,求解将哪些物品装⼊背包⾥物品价值总和最⼤。
例如:
vector weight = {1, 3, 4}; //各个物品的重量
vector value = {15, 20, 30}; //对应的价值int bagWeight = 4; //背包最大能放下多少重的物品
分析
1.确定dp数组
定义dp[i][j]为从下标为[0-i]的物品里任取,放进容量为j的背包,价值总和最大是多少
2、递推公式
-
当第i个物品放不下时:dp[i][j]=dp[i-1][j]
-
当第i个物品放得下时:
1、放第i个物品:dp[i][j]=dp[i-1][j-weight[i]]+value[i];
2、不放:dp[i][j]=dp[i-1][j]
3、初始化
由递推公式可以看出,i由i-1推导得出,j由j-weight[i]得出,所以dp[i][0]和dp[0][j]需要初始化。
dp[i][0]=0;
dp[0][j]为放入下标为0的物品时各容量背包能存放的最大价值
// 初始化 dp
vector<vector<int>> dp(weight.size() + 1, vector<int>(bagWeight + 1, 0));
for (int j = bagWeight; j >= weight[0]; j--) {
dp[0][j] = dp[0][j - weight[0]] + value[0];
}
4、遍历顺序
// weight数组的⼤⼩ 就是物品个数
for(int i = 1; i < weight.size(); i++) { // 遍历物品
for(int j = 0; j <= bagWeight; j++) { // 遍历背包容量
if (j < weight[i]) dp[i][j] = dp[i - 1][j];
变化
else dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
}
}
完整代码模板
// 0-1背包问题母代码(二维)
void bags()
{
vector<int> weight = {1, 3, 4}; //各个物品的重量
vector<int> value = {15, 20, 30}; //对应的价值
int bagWeight = 4; //背包最大能放下多少重的物品
// 二维数组:状态定义:dp[i][j]表示从0-i个物品中选择不超过j重量的物品的最大价值
vector<vector<int>> dp(weight.size() + 1, vector<int>(bagWeight + 1, 0));
// 初始化:第一列都是0,第一行表示只选取0号物品最大价值
for (int j = bagWeight; j >= weight[0]; j--)
dp[0][j] = dp[0][j - weight[0]] + value[0];
// weight数组的大小 就是物品个数
for (int i = 1; i < weight.size(); i++) // 遍历物品(第0个物品已经初始化)
{
for (int j = 0; j <= bagWeight; j++) // 遍历背包容量
{
if (j < weight[i]) //背包容量已经不足以拿第i个物品了
dp[i][j] = dp[i - 1][j];
else
dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);
}
}
cout << dp[weight.size() - 1][bagWeight] << endl;
}
5、优化
去除选取物品那一层 ,降为一维背包。为避免重复计算,需倒序遍历
void test_1_wei_bag_problem()
{
vector<int> weight = {1, 3, 4};
vector<int> value = {15, 20, 30};
int bagWeight = 4;
// 初始化
vector<int> dp(bagWeight + 1, 0);
for (int i = 0; i < weight.size(); i++)
{ // 遍历物品
for (int j = bagWeight; j >= weight[i]; j--)
{
dp[j] = max(dp[j], dp[j - weight[i]] + value[i]);
}
}
cout << dp[bagWeight] << endl;
}
参考链接:https://leetcode-cn.com/problems/last-stone-weight-ii/solution/yi-pian-wen-zhang-chi-tou-bei-bao-wen-ti-5lfv/