416. 分割等和子集

416. 分割等和子集

1.题目描述

给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
注意:
（1）每个数组中的元素不会超过 100
（2）数组的大小不会超过 200
示例 1:

示例 2:

2.思路（动态规划）

可以看成一个背包大小为 sum/2 的 0-1 背包问题。
1.状态定义：dp[i][j]表示从数组的 [0, i] 这个子区间内挑选一些正整数，每个数只能用一次，使得这些数的和恰好等于 j。
2.转移方程：考虑当前数据选与不选：
（1）选择 nums[i]，如果在 [0, i - 1] 这个子区间内就得找到一部分元素，使得它们的和为 j - nums[i]。
（2）不选择 nums[i]，如果在 [0, i - 1] 这个子区间内已经有一部分元素，使得它们的和为 j ，那么 dp[i][j] = true；
即状态转移方程是：dp[i][j] = dp[i - 1][j] or dp[i - 1][j - nums[i]]
因为第i个状态只与i-1有关，因此可以优化空间：dp[j] = dp[j] or dp[j-nums[i]]

3.代码

class Solution {
public:
    bool canPartition(vector<int>& nums) {
        if(nums.empty()){
            return false;
        }
        int sum = 0;
        for(auto &num : nums){
            sum += num;
        }
        if(sum & 1){//如果和为奇数，则不可能平分
            return false;
        }
        int W = sum / 2;
        vector<bool> dp(W+1,false);
        dp[0] = true;
        for(auto &num:nums){
            for(int j = W;j >= num;--j){
                dp[j] = dp[j] || dp[j-num];
            }
        }
        return dp[W];
    }

};

4.复杂度分析

时间复杂度：O(NC)：这里 N 是数组元素的个数，C 是数组元素的和的一半。
空间复杂度：O( C )