代码随想录打卡—day42—【DP】— 8.27 01背包基础

1 01背包基础

背包概述：

1.1 01背包是什么

有n件物品和一个最多能背重量为w 的背包。第i件物品的重量是weight[i]，得到的价值是value[i] 。每件物品只能用一次，求解将哪些物品装入背包里物品价值总和最大。

1.2 01背包二维数组

二维数组还比较好理解，五步法详见代码注释，AC代码：

#include<iostream>
using namespace std;

const int N = 1e3 + 10;

int weight[N];
int val[N];
int f[N][N];  // （i，j）表示只选取前i个商品 容积j的背包能够装的最大价值
/*

if(j < weight[i])f[i][j] = f[i-1][j];  //装不下就只有不装一条路
else f[i][j] = max(f[i-1][j],f[i-1][j-weight[i]]+val[i]);  //装的下就有装和不装两种选择
              不装i       装i

第0行--f[0][j]
for(int j = 0; j <= m ; j++)
{
    if(j >= weight[0])f[0][j] = val[0];
    else f[0][j] = 0;
}
第0列—— f[i][0]全设置为0

顺序：先商品i后背包j（先背包后商品也行）

*/

int main()
{
    int n,m;
    cin >> n >> m;
    
    for(int i = 0; i < n; i++)
    {
        int tmpw,tmpval;
        cin >> tmpw >> tmpval;
        weight[i] = tmpw;
        val[i] = tmpval;
    }
    
    //初始化
    for(int j = 0; j <= m ; j++)
        if(j >= weight[0])f[0][j] = val[0];
        else f[0][j] = 0;
        
    for(int i = 0; i <= n; i++)f[i][0] = 0;
    
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j++)
            if(j < weight[i])f[i][j] = f[i-1][j];  //装不下就只有不装一条路
            else f[i][j] = max(f[i-1][j],f[i-1][j-weight[i]]+val[i]);  //装的下就有装和不装两种选择

    //  for(int i = 0; i <= n; i++)
    // {
    //     for(int j = 0; j <= m; j++)
    //     {
    //         cout << f[i][j] << ' ';
    //     }
    //     puts("");
    // }
    
    cout << f[n][m];
    
    return 0;
}

1.3 01背包一维数组（滑动数组）

滑动数组实质就是利用01背包二维数组版本中每个f[i][j]只会用到上一层左侧 f[i-1][j-xx] 的数据，所以把上一层左侧数据保存在当前行的左侧，并且每次在当前行内遍历时候从右往左边遍历。AC代码：

#include<iostream>
using namespace std;

const int N = 1e3 + 10;

int weight[N];
int val[N];
int f[N];  // （j）表示容积j的背包能够装的最大价值

/*
转换方程：
if(j < weight[i])f[j] = f[j];  //装不下就只有不装一条路
else f[j] = max(f[j],f[j-weight[i]]+val[i]);  //装的下就有装和不装两种选择
              不装i       装i

第0行--f[0] = 0;

顺序：先商品i++后背包j--(j左边的作为上一层的记录),

*/

int main()
{
    int n,m;
    cin >> n >> m;
    
    for(int i = 0; i < n; i++)
    {
        int tmpw,tmpval;
        cin >> tmpw >> tmpval;
        weight[i] = tmpw;
        val[i] = tmpval;
    }
    
    //初始化
    for(int i = 0; i <= n; i++)
        for(int j = m; j >= 0; j--)
            if(j < weight[i])f[j] = f[j];  //装不下就只有不装一条路
            else f[j] = max(f[j],f[j-weight[i]]+val[i]);  //装的下就有装和不装两种选择


    
    cout << f[m];
    
    return 0;
}

2 01背包应用题1——416. 分割等和子集

416. 分割等和子集

一开始看到题目，想用贪心——排序+双指针 每次都把当前相对小的放进小的sum中，写完之后发现过不了：[1,1,2,2]这样的样例。错误代码：

class Solution {
public:
    /*
    左边的sum 小于 右边的sum l++，左边的sum+=
    左边的sum 大于 同理
    如果等于左边前进 
    1,2,3,4, 5,6,7,8,9, 10
    */
    bool canPartition(vector<int>& nums) 
    {
        // 解法1:排序+双指针
        if(nums.size() == 1)return 0;

        sort(nums.begin(),nums.end());

        int l = 0;
        int r = nums.size() - 1;
        int leftsum = 0;
        int rightsum = 0;
        while(l <= r)
        {
            if(leftsum <= rightsum)
            {
                leftsum += nums[l++];
            }
            else
            {
                rightsum += nums[r--];
            }
        }
        cout << l << "  " << r << endl;
        cout << leftsum << "  " << rightsum;
        if(leftsum == rightsum)return 1;
        else return 0;
    }
};

要明确本题中我们要使用的是01背包，因为元素我们只能用一次。

回归主题：首先，本题要求集合里能否出现总和为 sum / 2 的子集。

那么来一一对应一下本题，看看背包问题如何来解决。

只有确定了如下四点，才能把01背包问题套到本题上来。

• 背包的体积为sum / 2
• 背包要放入的商品（集合里的元素）重量为 元素的数值，价值也为元素的数值
• 背包如果正好装满，说明找到了总和为 sum / 2 的子集。
• 背包中每一个元素是不可重复放入。

具体分析过程见注释， AC代码：

class Solution {
public:
    // 找到一个背包 能够装nums.total（所有物体重量总和）/2的东西
    int dp[10005];    // 容积为i的背包 根据现有的物体重量情况最多能装的物体的重量
    /*
    转换成01背包问题：
    假设有一个nums.total/2的背包
    有若干个物体，每个物体的重量就是nums[i] 
    本题可以舍弃价值这个概念
    就是问一个nums.total/2的背包最多能够装的物体的重量是多少 能不能达到nums.total/2

    if(j < nums[i])dp[j] = dp[j];
    else dp[j] = max(dp[j] , dp[j - nums[i]]+nums[i]);

    dp[0] = 0;
    其他默认是0
    
    for物体i++ for容积j--

    模拟——

    */

    bool canPartition(vector<int>& nums) 
    {
        dp[0] = 0;

        int total = 0;
        for(auto i : nums)total += i;
        
        if(total % 2 == 0)total /= 2;
        else return 0;

        for(int i = 0; i < nums.size();i++)
        {
            for(int j = total ; j >= 0; j--)
            {
                if(j < nums[i])dp[j] = dp[j];
                else dp[j] = max(dp[j] , dp[j - nums[i]]+nums[i]);
            }
        }

        if(dp[total] == total)return 1;
        else return 0;

    }
};