题目描述
给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
- 每个数组中的元素不会超过 100
- 数组的大小不会超过 200
来源,leetcode 每日一题 416. 分割等和子集
例如:
输入: [1, 5, 11, 5]
输出: true
解释: 数组可以分割成 [1, 5, 5] 和 [11].
解题思路
- 当数组的元素和为奇数的时候,那么它肯定不能等分,排除
- 当数组的元素最大值>的时候,意味着,除了最大值以外,其他所有元素和加起来要小于总和的一半,显然也不可能,排除
- 其余情况,则是很常见的dp思路
dp,每个位置对于值j 可以有两个选择,选或者不选。 如果选,那么这个位置的结果是什么? result[i][j] = result[i-1][j-nums[i]] ->这个的解释: j = (j-nums[i]) + nums[i] 当前位置成立的前提条件是,j-nums[i]要成立。 如果不选,这个位置的结果是什么?(当此处的值>j的时候是一定不选的) result[i][j] = result[i-1][j]
依次类推,当j == target的时候,就是结果
代码
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum = 0;
int n = nums.size();
if (n < 2) {
return false;
}
int maxNum = 0;
for (int i = 0; i < n; i++) {
sum += nums[i];
maxNum = (maxNum >= nums[i] ? maxNum : nums[i]);
}
if (sum % 2 != 0) {
return false;
}
int target = sum / 2;
if (maxNum > target) {
return false;
}
vector<vector<int>> result(n+1, vector<int>(target+1, 0));
for (int i = 0; i < n; i++) {
result[i][0] = true;
}
result[0][nums[0]] = true;
for (int i = 1; i < n; i++) {
for (int j = 1; j < target + 1; j++) {
if (j > nums[i]) {
result[i][j] = result[i-1][j] | result[i-1][j-nums[i]];
} else {
result[i][j] = result[i-1][j];
}
}
}
return result[n-1][target];
}
};