题目描述
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
- 每个数组中的元素不会超过 100
- 数组的大小不会超过 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];
}
};