代码随想录算法训练营第四十六天 | LeetCode 1049、494、474

---

前言

LeetCode题目：LeetCode 1049、494、474
Takeaway：背包问题——01背包问题，如何把别的问题转化为01背包问题

---

一、1049

本题的递推公式是 dp[j] = max(dp[j-stones[i]]+stones[i], dp[j])；让分成的两堆尽可能一样大就符合题意了。

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        vector<int> dp(15001, 0);
        dp[0] = 0;
        int sum = 0;

        for(int i=0; i<stones.size(); i++){
            sum += stones[i];
        }

        int target = sum/2;

        for(int i=0; i<stones.size(); i++){
            for(int j=target; j>=stones[i]; j--){
                dp[j] = max(dp[j-stones[i]]+stones[i], dp[j]);
            }
        }

        return sum - dp[target] - dp[target];
    }
};

---

二、494

本题的递推公式是 dp[j] += dp[j-nums[i]]；这可以从定义来反向分析一下，就是已经拿到一个物品之后，剩下的部分怎么拿，就是有dp[j-nums[i]]种方法,累加到一起就是dp[j] 。

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

---

三、474

本题的递推公式是 dp[i][j] = max(dp[i][j], dp[i - zeroNum][j - oneNum] + 1)；数组是2维，但是这也是压缩后的数组，第三维度-物品被压缩，现在的[i][j]其实就是原来的j

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

---

总结

背包问题——01背包问题，转化一下问题变成01背包即可。