LeetCode 416. 分割等和子集

题目描述

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

注意:

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

示例 1:

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

输出: true

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

示例 2:

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

输出: false

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

问题分析

先将给定数组nums的数全部加起来，判断如果sum为奇数，那么返回false。然后我们把和sum除以2，记为target。然后创建一个二维dp数组，行数为nums元素个数n，列数为target + 1。dp[i][j]表示数组的前 i+1 个元素中，能否得到和为 j 的子集。dp数组的第0行只有dp[0][0]和dp[0][nums[0]]是true，其他都是默认的初始值false。dp数组的第0列除了dp[0][0]，其余都是默认值false，所以我们从第1行第1列开始计算dp数组，如果在计算dp[i][j]时，j >= nums[i]，那么分为子集包括nums[i]和不包括nums[i]的情况，所以dp[i][j] = dp[i-1][j] || dp[i-1][j-nums[i]]; 不包括的情况和包括的情况中，有一个为true，那么dp[i][j]就为true。如果在计算dp[i][j]时，j < nums[i]，那么只有子集不包括nums[i]的情况了，所以dp[i][j] = dp[i-1][j]; 如此进行，在dp数组全部计算结束后，返回dp[n-1][target]。

代码实现

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