给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
提示:
1 <= nums.length <= 200
1 <= nums[i] <= 100
解题思路:
数组分割成两个⼦集,使得两个⼦集的元素和相等。 只要找到集合⾥能够出现 sum / 2 的⼦集总和,就算是可以分割成两个相同元素和⼦集了。
代码:
class Solution {
public:
bool canPartition(vector<int>& nums) {
// 题目中说:每个数组中的元素不会超过 100,数组的⼤⼩不会超过 200
// 总和不会⼤于20000,背包最⼤只需要其中⼀半,所以10001⼤⼩就可以了
vector<int> dp(10001, 0);
//dp[i]表示 背包总容量是i,最⼤可以凑成i的⼦集总和为dp[i]。
int sum = 0,target = 0;
dp[0] = 0;
for(int i=0;i<nums.size();i++){
sum += nums[i];
}
if(sum % 2 == 1) return false;
target = sum / 2;
//01背包
for(int i=1;i<nums.size();i++){
for(int j = target; j>=nums[i];j--){// 每⼀个元素⼀定是不可重复放⼊,所以从⼤到⼩遍历
dp[j] = max(dp[j],dp[j-nums[i]]+nums[i]);
}
}
if(target == dp[target]) return true;
return false;
}
};