LeetCode 416. 分割等和子集

题目描述：

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

注意:

1. 每个数组中的元素不会超过 100
2. 数组的大小不会超过 200

示例 1:

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

输出: true

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

 

示例 2:

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

输出: false

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

解题思路：

1. 递归，将数组从大到小排序（从小到大排序递归会超时），凑sum /2 的值。

代码如下：

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = accumulate(nums.begin(),nums.end(),0);
        if(sum % 2 != 0) return false;
        int target = sum / 2;
        sort(nums.begin(), nums.end(),greater<int>());
        return dfs(nums, target, 0);
    }
    bool dfs(vector<int>& nums,int target, int start){
        if(target == 0) return true;
        if(nums[start] > target) return false;
        for(int i = start; i < nums.size() && nums[i] <= target; ++i){
            if(dfs(nums, target - nums[i], i + 1))
                return true;
        }
        return false;
    }
};

2. 动态规划， 设置一个vector<int> dp 来记录数组各元素相加的累计和，判断是否等于sum/2。

代码如下：

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