416. Partition Equal Subset Sum 题解

416. 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.

题目链接：416. Partition Equal Subset Sum

算法描述：

        由题意可知，给定一个元素全部为正的数组，我们将确定这个数组是否可以分为两个子数组，其中每个子数组的和相同。

             首先我们能够确定的是，如果原数组的和为奇数，则原数组不可能划分为两个和相同的子数组。如果和为偶数，则和的一半就是每个子数组和的值 “target”。

         之后，我们用动态规划来做这道题，我们定义一个 bool 类型的数组 “ans[target + 1]” ，其中 “ans[0] = true”，其余为 “false”，“ans[i]” 表示 “i” 是否可以表示成原数组的子数组的和。

          我们开始遍历原数组中的每个元素，当遍历到元素 “nums[i]” 时，我们需要更新 “ans[]" 数组，对于 "nums[i]” 到 “target” 之间的值 “j”，如果 “ans[j - nums[i]] ” 为 “true” ，那么 “ans[j]” 就是 “true”。

代码：

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