力扣题目链接:力扣
题目描述:
给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
示例 1:
输入:nums = [1,5,11,5]
输出:true
解释:数组可以分割成 [1, 5, 5] 和 [11] 。
示例 2:
输入:nums = [1,2,3,5]
输出:false
解释:数组不能分割成两个元素和相等的子集。
解答:
类似背包问题
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum=0;
if(nums.size()==0)return false;
for(int i=0;i<nums.size();++i){
sum+=nums[i];
}
if(sum%2!=0)return false;
sum=sum/2;
vector<int> dp(sum+1);//空间优化,滚动数组,dp[j]=1表示存在和为j的集合
for(int i=0;i<nums.size();++i){
for(int j=sum;j>nums[i];--j){//为防止同数用多次,从后往前遍历
dp[j]=max(dp[j-nums[i]],dp[j]);
}
if(nums[i]<=sum){//初始值
dp[nums[i]]=1;
}
}
if (dp[sum] == 1) return true;
return false;
}
};