欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

416. 分割等和子集

程序员文章站 2022-07-15 16:19:40
...

Q:

416. 分割等和子集

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);
    }
};