背包问题学习记录

背包问题

动态规划基本步骤

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/