代码随想录算法训练营第四十二天|416.分割等和子集、1049.最后一块石头的重量 II

文档链接：https://programmercarl.com/

LeetCode416.分割等和子集

题目链接：https://leetcode.cn/problems/partition-equal-subset-sum/

思路：不同于传统的01背包问题求解的容量不超过j的最大价值，这题是求背包总价值等于给定子集和的一半。可以把重量和价值看成一样的。

动规：

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

---

LeetCode1049.最后一块石头的重量 II

题目链接：https://leetcode.cn/problems/last-stone-weight-ii/

思路：跟上一题一样，不过更难想，要把石头分成尽可能同样重的两堆，然后作差便是最小石头。

动规：

class Solution {
public:
    int lastStoneWeightII(vector<int>& stones) {
        int sum = 0;
        for(int i : stones) sum += i;
        int targrt = sum / 2;
        vector<int> dp(targrt + 1, 0);
        dp[0] = 0;
        for(int i = 0; i < stones.size(); i++) {
            for(int j = targrt; j >= stones[i]; j--) {
                dp[j] = max(dp[j], dp[j - stones[i]] + stones[i]);
            }
        }
        int result = sum - dp[targrt] - dp[targrt];
        return result;
    }
};

总结：说实话很难看出是01背包。