Leetcode 416：分割等和子集

题目描述

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

注意:

每个数组中的元素不会超过 100
数组的大小不会超过 200
示例 1:

输入: [1, 5, 11, 5]

输出: true

解释: 数组可以分割成 [1, 5, 5] 和 [11].
 

示例 2:

输入: [1, 2, 3, 5]

输出: false

解释: 数组不能分割成两个元素和相等的子集.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/partition-equal-subset-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

二维背包

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

一维背包

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