给定一个只包含正整数的非空数组。是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
注意:
- 每个数组中的元素不会超过 100
- 数组的大小不会超过 200
示例 1:
输入: [1, 5, 11, 5] 输出: true 解释: 数组可以分割成 [1, 5, 5] 和 [11].
解题思路
1. 递规实现(自顶向下)
class Solution {
private:
//考虑[0...index]能否恰好放入容量为sum的背包中
bool tryPartition(const vector<int>&nums,int index,int sum){
if(sum==0)
return true;
//sum==0就说明背包没有任何空间了,恰好填存满了背包
if(index<0||sum<0)
return false;
//index<0,说明没有物品可以装入背包,且背包没有装满
//sum<0,说明当前物品太大,超出了背包可容纳的空间
return tryPartition(nums,index-1,sum)||tryPartition(nums,index-1,sum-nums[index]);
}
public:
bool canPartition(vector<int>& nums) {
int sum=0;
for(int i=0;i<nums.size();i++){
assert(nums[i]>0);
sum+=nums[i];
}
if(sum%2!=0)
return false;
return tryPartition(nums,nums.size()-1,sum/2);
}
};
上述代码超出了leetcode 时间限止
2. 递规+记忆化搜索实现(自顶向下)
class Solution {
private:
//考虑[0...index]能否恰好放入容量为sum的背包中
bool tryPartition(const vector<int>&nums,int index,int sum,vector<vector<int>> &memo){
if(sum==0)
return true;
//sum==0就说明背包没有任何空间了,恰好填存满了背包
if(index<0||sum<0)
return false;
//index<0,说明没有物品可以装入背包,且背包没有装满
//sum<0,说明当前物品太大,超出了背包可容纳的空间
if(memo[index][sum]!=-1)
return memo[index][sum]==1;//或者返回true
memo[index][sum]=(tryPartition(nums,index-1,sum,memo)||tryPartition(nums,index-1,sum-nums[index],memo))?1:0;
return memo[index][sum]==1;//(直接返回memo[index][sum]也可,使用了类型自动转化)
}
public:
bool canPartition(vector<int>& nums) {
int sum=0;
for(int i=0;i<nums.size();i++){
assert(nums[i]>0);
sum+=nums[i];
}
if(sum%2!=0)
return false;
vector<vector<int>>memo(nums.size(),vector<int>(sum/2+1,-1));
return tryPartition(nums,nums.size()-1,sum/2,memo);
}
};
动态规化(自底向上)c++ 代码
class Solution {
public:
bool canPartition(vector<int>& nums) {
int sum=0;
for(int i=0;i<nums.size();i++){
assert(nums[i]>0);
sum+=nums[i];
}
if(sum%2!=0)
return false;
int n=nums.size();
int C=sum/2;
vector<bool>memo(C+1,false);
for(int j=0;j<=C;j++)
memo[j]=(j==nums[0]);
//0号物品能否将背包填满
for(int i=1;i<n;i++)
for(int j=C;j>=nums[i];j--)
memo[j]=memo[j]||memo[j-nums[i]];
return memo[C];
}
};
作业题:
322、377、474、139、494