代码随想录第四十三天——目标和,一和零

leetcode 494. 目标和

题目链接:目标和

本题要如何使表达式结果为target:

既然为target,那么就一定有 left组合 - right组合 = target

left + right = sum,而sum是固定的。right = sum - left,所以left - (sum - left) = target 推导出 left = (target + sum)/2 。

此时问题就转化为:装满容量为left的背包,有几种方法

本题是求解装满一个背包有几种方法,属于组合问题,与之前的背包问题不同。

  1. 确定dp数组以及下标的含义
    dp[j] :填满j(包括j)这么大容积的包,有dp[j]种方法
  2. 确定递推公式
    以dp[j],j 为5为例:
    已经有1(nums[i])的话,有 dp[4]种方法凑成容量5的背包
    已经有2(nums[i])的话,有 dp[3]种方法凑成容量5的背包
    已经有3(nums[i])的话,有 dp[2]中方法凑成容量5的背包
    已经有4(nums[i])的话,有 dp[1]中方法凑成容量5的背包
    已经有5(nums[i])的话,有 dp[0]中方法凑成容量5的背包
    只要有了nums[i],凑成dp[j]就有dp[j - nums[i]] 种方法,凑成dp[5]就是把所有的dp[j - nums[i]] 累加起来。所以求组和类的递归公式: dp[j] += dp[j - nums[i]]
  3. dp数组初始化
    dp[0]初始化为1
  4. 确定遍历顺序
    nums放在外循环,target在内循环,且内循环倒序
cpp 复制代码
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 (abs(target) > sum) return 0; // 此时没有方案
        if ((target + sum) % 2 == 1) return 0; // 此时没有方案
        int bagSize = (target + sum) / 2;
        vector<int> dp(bagSize + 1, 0);
        dp[0] = 1;
        for (int i = 0; i < nums.size(); i++) {
            for (int j = bagSize; j >= nums[i]; j--) {
                dp[j] += dp[j - nums[i]];
            }
        }
        return dp[bagSize];
    }
};

时间复杂度:O(n × m),n为正数个数,m为背包容量

空间复杂度:O(m),m为背包容量

leetcode 474. 一和零

题目链接:一和零
本题中strs 数组里的元素就是物品,每个物品都是一个!
m 和 n相当于是一个背包,两个维度的背包

  1. dp数组及下标的含义
    dp[i][j]:最多有i个0和j个1的strs的最大子集的大小为dp[i][j]。
  2. 确定递推公式
    dp[i][j] 可以由前一个strs里的字符串推导出来,strs里的字符串有zeroNum个0,oneNum个1。
    dp[i][j] 就可以是 dp[i - zeroNum][j - oneNum] + 1。然后在遍历的过程中,取dp[i][j]的最大值。
    所以递推公式:dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1);
    字符串的zeroNum和oneNum相当于物品的重量(weight[i]),物品的重量有两个维度。字符串本身的个数相当于物品的价值(value[i])。
  3. dp数组初始化
    物品价值不会是负数,初始为0,保证递推的时候dp[i][j]不会被初始值覆盖。
  4. 遍历顺序
    外层for循环遍历物品,内层for循环遍历背包容量且从后向前遍历。
cpp 复制代码
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);
        }
    }
}

整体代码如下:

cpp 复制代码
class Solution {
public:
    int findMaxForm(vector<string>& strs, int m, int n) {
        vector<vector<int>> dp(m + 1, vector<int> (n + 1, 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];
    }
};

时间复杂度: O(kmn),k 为strs的长度

空间复杂度: O(mn)

相关推荐
摆烂仙君1 小时前
小米2025年校招笔试真题手撕(二)
算法
xin007hoyo5 小时前
算法笔记·数学·最大公约数
笔记·算法
丶Darling.6 小时前
Day126 | 灵神 | 二叉树 | 层数最深的叶子结点的和
数据结构·c++·算法·二叉树·深度优先
小O的算法实验室6 小时前
2025年ESWA SCI1区TOP,离散人工蜂群算法+多农场多除草机器人任务分配,深度解析+性能实测
算法·智能算法改进
2401_859049087 小时前
Git使用
arm开发·git·stm32·单片机·mcu·算法
冠位观测者8 小时前
【Leetcode 每日一题】2942. 查找包含给定字符的单词
算法·leetcode·职场和发展
蓝心湄8 小时前
C语言-枚举
c语言·开发语言·算法
轮到我狗叫了8 小时前
力扣小题, 力扣113.路径总和II力扣.111二叉树的最小深度 力扣.221最大正方形力扣5.最长回文子串更加优秀的算法:中心扩展算法
算法·leetcode·深度优先
黎明smaly8 小时前
【C语言】复习~数组和指针
c语言·开发语言·数据结构·c++·leetcode
haven-8529 小时前
最大似然估计(Maximum Likelihood Estimation, MLE)详解
算法·机器学习·概率论