代码随想录算法训练营第三十六天|01背包问题 二维 ,01背包问题 一维 ,416. 分割等和子集

背包理论基础

01 背包(二维)

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

背包最大重量为4。

物品为:

重量价值
物品0115
物品1320
物品2430

问背包能背的物品最大价值是多少?

思路:依然是动态规划。

解决:动态规划五步曲

        1.确定dp数组及下标含义;

        dp[i][j] 表示从下标为[0-i]的物品里任意取,放进容量为j的背包,价值总和最大是多少。

        2.确定递推公式;

                考虑dp[i][j]是如何推导过来;

                ①加入第i件物品;

                        加入第i件物品后,相较于第i-1个物品,加入第i件物品后价值总和就等于dp[i-1][j-weight[i]]+value[i]

                ②第i件物品加入重量超标,不加入;

                        不加入价值总和就是放入第i-1个物品后价值总和:dp[i-1][j]

        所以dp[i][j]=max(dp[i-1][j],dp[i-1][j-weight[i]]+value[i])

        3.确定dp数组初始化;

        首先当j=0时,也就是书包最大重量为0时,dp[i][0]=0;其次只放物品0时,背包价值总和都是15,所以dp[0][j]=15(j>weight[0])。注意这里容量要大于第一个物品的重量。

        4.确定遍历顺序;

        遍历顺序肯定是从左向右,从上到下,最后遍历到右下角。

        5.举例推导dp数组。

        

代码:

void test_2_wei_bag_problem1() {
    vector<int> weight={1,3,4};
    vector<int> value={15,20,30};
    int bagweight = 4;
    vector<vector<int>> dp(weight.size(),vector<int>(bagweight+1,0));//i是物品数量,j是背包容量
    //初始化
    for(int j=weight[0];j<=bagweight;j++){
        dp[0][j]=value[0];
    }
// 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]);

        }
    }

    cout << dp[weight.size() - 1][bagweight] << endl;
int main() {
    test_2_wei_bag_problem1();
}

 01 背包(一维)

例子:背包最大重量为4。

物品为:

重量价值
物品0115
物品1320
物品2430

问背包能背的物品最大价值是多少?

思路:对于背包问题其实状态都是可以压缩的。

在使用二维数组的时候,递推公式:dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]);

其实可以发现如果把dp[i - 1]那一层拷贝到dp[i]上,表达式完全可以是:dp[i][j] = max(dp[i][j], dp[i][j - weight[i]] + value[i]);

与其把dp[i - 1]这一层拷贝到dp[i]上,不如只用一个一维数组了,只用dp[j](一维数组,也可以理解是一个滚动数组)。

解决:动态规划五步曲

        1.确定dp[j]的含义;

         dp[j]表示容量为j的背包,能背的最大价值。

        2.确定递推公式;

        dp[j]=max(dp[j],dp[j-weight[i]]+value[i]);这里还是一样可以选择放i物品或者不放。

        3.确定dp初始化;

        最开始j=0时候,装不了物品。dp[0]=0;

        4.确定遍历顺序;

        举一个例子:物品0的重量weight[0] = 1,价值value[0] = 15

        如果正序遍历

        dp[1] = dp[1 - weight[0]] + value[0] = 15

        dp[2] = dp[2 - weight[0]] + value[0] = 30

        此时dp[2]就已经是30了,意味着物品0,被放入了两次,所以不能正序遍历。        

        倒序就是先算dp[2]

        dp[2] = dp[2 - weight[0]] + value[0] = 15 (dp数组已经都初始化为0)

        dp[1] = dp[1 - weight[0]] + value[0] = 15

        所以从后往前循环,每次取得状态不会和之前取得状态重合,这样每种物品就只取一次了

        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;
}

int main() {
    test_1_wei_bag_problem();
}

416. 分割等和子集 - 力扣(LeetCode)

给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。

示例 1:

输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。

示例 2:

输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。

思路:这个问题如何转化成背包问题?最好的结果就是分成两个子集和相等,既然和相等,那么子集的和就是总和的一半,也就是说,从nums数组种取出元素,能够满足和等于总和的一半,那么结果就是true。那这里背包容量就是总和的一半,物品的价值就是元素数值,物品重量可以说每个都是1,也可以不用管

解决:动态规划五步曲:

        1.确定dp[i]的含义;

        j表示背包总容量,这里相当于目标值,从目标值递减遍历;背的最大重量为dp[j],当dp[j]等于目标值时说明可以分割成两个子集。

        2.确定递推公式;

        dp[j] = max(dp[j], dp[j - nums[i]] + nums[i])。

        3.确定dp初始化;

        初始化和一维数组一样,都设置为0。

        4.确定遍历顺序;

        遍历顺序从大到小,防止重复放入。

        5.举例推导公式。

代码:

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum=0;
        for(int i=0;i<nums.size();i++){
            sum+=nums[i];
        }
        vector<int> dp(10001, 0);
        if(sum%2==1){
            return false;
        }
        int target=sum/2;
        for(int i=0;i<nums.size();i++){
            for(int j=target;j>=nums[i];j--){//注意遍历顺序
                dp[j]=max(dp[j],dp[j-nums[i]]+nums[i]);
            }
        }
        if (dp[target] == target) return true;
        return false;
    }
};

  • 20
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值