39. Combination Sum & 40. Combination Sum II

最新推荐文章于 2020-07-31 05:05:06 发布

cColdTea

最新推荐文章于 2020-07-31 05:05:06 发布

阅读量259

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/emmaczw/article/details/80399201

版权

leetcode 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

这题很简单，主要是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++这一步操作）