416. 分割等和子集
程序员文章站
2022-07-15 16:19:40
...
Q:
A:
1.DP
主要题目中说了不超过100个数字,数字都不超过200。所以可能的和不会超过20000,这个量级对计算机来说不算大,所以考虑用DP考察每个可能的和是否存在。
class Solution {
public:
bool canPartition(vector<int>& nums) {
if(nums.empty()){
return false;
}
int sum=accumulate(nums.begin(),nums.end(),0);
if (sum&1){ //和奇数,无法一分为二
return false;
}
int siz=nums.size();
vector<bool> dp(sum/2+1,false);
for(int i=0;i<sum/2+1;++i){
dp[i]=nums[0]==i;
}
for(int i=1;i<siz;++i){
for(int j=sum/2;j>=0;--j){
if(nums[i]<=j){
dp[j]=dp[j] or dp[j-nums[i]];
}
}
}
return dp.back();
}
};
2.DFS
偷懒截的别人的
要注意剪枝,并且先排序,要从大的开始选。如果从小的开始选会多很多可能。
class Solution {
public:
bool dfs(vector<int>& num, int pos, int left){
if (left == 0)return true;
if (pos < 0)return false;
if (left < num[pos])return false;
bool ret = dfs(num, pos - 1, left - num[pos]);
if (ret)return ret;
return dfs(num, pos - 1, left);
}
bool canPartition(vector<int>& nums) {
int sum = accumulate(nums.begin(), nums.end(), 0);
if (sum &1) return false;
sort(nums.begin(), nums.end());
//printf ("%d", sum);
return dfs(nums, nums.size() - 1, sum/2);
}
};