【LeetCode】416. Partition Equal Subset Sum

416. Partition Equal Subset Sum

Description：
Given a non-empty array containing only positive integers, find if the array can be partitioned into two subsets such that the sum of elements in both subsets is equal.
Note:
Each of the array element will not exceed 100.
The array size will not exceed 200.
Difficulty：Medium

Example:

Input: [1, 5, 11, 5]

Output: true

Explanation: The array can be partitioned as [1, 5, 5] and [11].

方法1:转化问题，动态规划

• Time complexity : $O\left(n\ast sum\right)$
• Space complexity : $O\left(sum/2\right)$
  思路：
  因为先做了494. Target Sum 这道题目，所以自然而然想到这道题目可以转化为添加正负号使得sum为0，再接着转化就是在nums中找到一个集合，使得这个集合的和为sum/2。
  由于这种思维惯性，导致了运算操作冗余，其实没必要计算出来有几种组合方式，只需要记录是否有这种组合即可，具体代码请看方法2。

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        if(sum % 2 == 1) return false;
        int s = sum / 2;
        vector<double> dp(s+1); //这里用了double是因为需要记录的次数实在太大，只能用double，所以这个方法不可取
        dp[0] = 1;
        for( auto num : nums)
            for(int i = s;i >= num; i--)
                dp[i] += dp[i-num];
        return dp[s] > 0 ? true : false;
    }
};

方法2：转化问题，动态规划

• Time complexity : $O\left(n\right)$
• Space complexity : $O\left(sum\right)$
  思路：因为只需要记录是否存在这种可能，所以用bitset即可。
  遍历数组，每次将bits向左移num位，并且与自己与，因为num的加入，bits的某些位发生了变化，保持了累加性。说的比较难懂，来点例子。

核心代码：bits |= bits << num;
nums = [1, 1, 2, 2]
bits = 0000001
-->1 bits = 0000011
-->1 bits = 0000111
-->2 bits = 0011111
-->2 bits = 1111111

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        int sum = accumulate(nums.begin(), nums.end(), 0);
        bitset<10001> bits(1);
        for(auto num : nums)
            bits |= bits << num;
        return !(sum & 1) && bits[sum / 2];
    }
};