416. 分割等和子集/C++

在这里插入图片描述

动态规划,一维版本
dp[i][j]表示是否存在[0,i]中取一些元素使其和为j
dp[i][j] = dp[i-1][j] || dp[i-1][s-nums[i]](分别表示取i与不取i的情况)
然后第i行的值只依赖于第i-1行,因此可以轻易缩减为2行,
如果从右向左求值,则只需1行。

bool canPartition(vector<int>& nums) {
    if(nums.size()==0||nums.size()==1)
        return false;
    
    int sum=accumulate(nums.begin(),nums.end(),0);
    if(sum%2) return false;
    
    int n = nums.size();
    int target = sum/2;
    vector<bool> dp(target+1,false);
    dp[0] = true;
    
    for(int i=0;i<n;++i){
        for(int j=target;j>=nums[i];--j){
            dp[j] = dp[j] || dp[j-nums[i]];
        }
    }
    return dp[target];
}

剪枝,更快

class Solution {
public:
    int target;
    bool check(vector<int> &nums, int cur1,int cur2,int index){
        //剪枝,如果有一个和超过了target,则停止搜索(这个剪枝非常重要)
        if(target < cur1 || target < cur2 || index == -1) {
            return false;
        }
        if(target == cur1||target == cur2) {
            return true;
        }
        //将nums[index]分配cur1或者cur2
        return check(nums, cur1 + nums[index], cur2, index-1) 
        		|| check(nums, cur1, cur2 + nums[index], index-1);
    }
    
    bool canPartition(vector<int>& nums) {
        //accumulate带有三个形参:头两个形参指定要累加的元素范围,第三个形参则是累加的初值。
        int sum=accumulate(nums.begin(),nums.end(),0);
        if(sum % 2 != 0){
            return false;
        }
        target = sum / 2;
        return check(nums, 0, 0, nums.size() - 1);
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值