39. Combination Sum & 40. Combination Sum II

这题很简单,主要是push、pop会花比较多的时间,做个预判,不要所有的都先push,80%多吧。

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> ans_temp;
    
    void dfs(vector<int>& candidates, int index, int target, int sum){
        if(sum == target) ans.push_back(ans_temp);
        if(sum > target) return;
        for(int i=index; i<candidates.size(); i++){
            if(sum+candidates[i] > target) continue;
            ans_temp.push_back(candidates[i]);
            dfs(candidates, i, target, sum+candidates[i]);
            ans_temp.pop_back();
        }
    }
    
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        for(int i=0; i<candidates.size(); i++){
            ans_temp.push_back(candidates[i]);
            dfs(candidates, i, target, candidates[i]);
            ans_temp.pop_back();
        }
        return ans;
    }
};

于是顺便把40也做了,这个是不能允许重复的,一开始的思路是用set,再用copy转成set

vector<vector<int>> ans(set_ans.size()); //这里需要先定义一个空间,不然copy会报错,具体注意事项参照链接
copy(set_ans.begin(), set_ans.end(), ans.begin());

即可,但是有点费时间,于是想了如下代码:

class Solution {
public:
    vector<vector<int>> ans;
    vector<int> ans_temp;
    
    void dfs(vector<int>& candidates, int index, int target, int sum){
        if(sum == target) ans.push_back(ans_temp);
        if(sum > target) return;
        for(int i=index+1; i<candidates.size();){
            ans_temp.push_back(candidates[i]);
            dfs(candidates, i, target, sum+candidates[i]);
            ans_temp.pop_back();
            do{i++;}
            while(i<candidates.size() && candidates[i]==candidates[i-1]);
        }
    }
    
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(), candidates.end());
        for(int i=0; i<candidates.size(); ){
            ans_temp.push_back(candidates[i]);
            dfs(candidates, i, target, candidates[i]);
            ans_temp.pop_back();
            do{i++;}
            while(i<candidates.size() && candidates[i]==candidates[i-1]);
        }
        return ans;
    }
};

即在准备push下一个数字的时候,需要判断一下是否与上一个相同(这里要先sort,也很费时间),若相同,则其不管怎么挑选之后的数字都会与前一个相同的数字挑选出的组合有重复集合,此时则需要跳过该数字(i++这一步操作)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值