[LeetCode OJ]Partition Equal Subset Sum

【问题描述】

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:

1. Each of the array element will not exceed 100.
2. The array size will not exceed 200.

Example 1:

Input: [1, 5, 11, 5]

Output: true

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

Example 2:

Input: [1, 2, 3, 5]

Output: false

Explanation: The array cannot be partitioned into equal sum subsets.

问题来源：partition-equal-subset-sum

【解题分析】

题意是让我们把一个数组划分成两个子数组，要求两个子数组的和相等。

假设原来数组的和为sum，数组大小为n。那么我们不妨这样假设，从原来的数组中挑选i个数(i≤n)，使得它们的和为sum/2，这样便成了很经典的0/1背包问题。

状态表示：dp[i][j]代表前i个数中和为j的方案是否可行，0代表不可行，1代表可行。

状态转移方程：dp[i][j] |= dp[i-1][j] + dp[i-1][j-nums[i]]，dp[0][0] = 1

返回结果：dp[n][sum/2]

我们再对它进行优化：

我们对它进行降维处理，从dp[i][j]降成dp[j]，循环n次去更新dp[j]。

状态转移方程：dp[j] |= dp[j-nums[i]]，i是当前循环的次数，i从0到n-1。

返回结果：dp[sum/2]

时间复杂度O(N^2)，空间复杂度O(N)。

【源代码】

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