DAY43:背包问题提升1049、494、474

Leetcode: 1049 最后一块石头的重量 II

这道题和昨天的最后一道题很像,都是重量和价值一样等于stone[i]。

本质思想是尽量将石头分成相似的两堆。如果出现两堆价值一样,那==0,如果不一样,就用大的那堆减去小的那堆就是最后相撞之后的重量。从而把题目化简为01背包问题。

时间复杂度:O(m × n)

空间复杂度:O(m)

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int result = 0;//需要初始化,没有初始化为0会报错
        for(int i = 0; i < stones.size(); i++){
            result += stones[i];
        }

        int target = result / 2;//分为两堆石块的目标
        vector<int> dp(target + 1, 0);//确定dp数组的大小
        for(int i = 0; i < stones.size(); i++){
            for(int j = target; j >= stones[i]; j--){//背包问题
                dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
            }
        }
        return result - dp[target] - dp[target];//因为tatget = result/2向下取整,这样result-dp[target]肯定比dp[target]大

    }
};

Leetcode: 494 目标和

假设加法的总和为x,那么减法对应的总和就是sum - x,我们要求的是 x - (sum - x) = target,所以x = (target + sum) / 2。此时问题就转化为,装满容量为x的背包,有几种方法。这里注意/是向下取整,所以如果出现(target+sum)是奇数,是没有解的。

1、下标和定义

dp[j] 表示:填满j(包括j)这么大容积的包,有dp[j]种方法。

2、递推公式

只要搞到nums[i],凑成dp[j]就有dp[j - nums[i]] 种方法。所以需要将所有的 dp[j - nums[i]] 累加起来。

3、初始化

初始化dp[0]=1这个很重要,如果初始化为0,就会出错因为所有的结果都是由0导出的,所以需要初始化为1,保证有一个方案

时间复杂度:O(m × n)

空间复杂度:O(m)

class Solution {
public:
    int findTargetSumWays(vector<int>& nums, int target) {
        int sum = 0;
        for(int i = 0; i < nums.size(); i++){
            sum += nums[i];
        }
        if(sum < abs(target)) return 0;//如果目标比总和大,肯定无法达到
        if((sum + target) % 2 == 1) return 0;//如果出现奇数,就有非整数

        int result = (sum + target) / 2;
        vector<int> dp(result + 1, 0);
        dp[0] = 1;
        for(int i = 0; i < nums.size(); i++){//注意从0的情况开始
            for(int j = result; j >= nums[i]; j--){
                dp[j] += dp[j - nums[i]];
            }
        }
        return dp[result];
    }
};

在求装满背包有几种方法的情况下,递推公式一般为:

dp[j] += dp[j - nums[i]];

Leetcode: 474 一和零

这道题主要是题目看不懂,思路也很难想。

参考

代码随想录

1、确定下标含义

dp[i][j]:最多有i个0和j个1的strs的最大子集的大小为dp[i][j]。

2、递归公式

相当于这里是一个有两个维度的01背包问题。

dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);

字符串的zeroNum和oneNum相当于物品的重量(weight[i]),字符串本身的个数相当于物品的价值(value[i])。

3、递归逻辑

01背包一定是外层for循环遍历物品,内层for循环遍历背包容量且从后向前遍历!物品就是strs里的字符串,背包容量就是题目描述中的m和n。

时间复杂度为O(KMN)

空间复杂度为O(MN)

class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m + 1, vector<int> (n + 1, 0)); // 默认初始化0
        for (string str : strs) { // 遍历物品
            int oneNum = 0, zeroNum = 0;
            for (char c : str) {
                if (c == '0') zeroNum++;
                else oneNum++;
            }
            for (int i = m; i >= zeroNum; i--) { // 遍历背包容量且从后向前遍历!
                for (int j = n; j >= oneNum; j--) {
                    dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
                }
            }
        }
        return dp[m][n];
    }
};
  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值