LeetCode40. Combination Sum II

题目链接:

https://leetcode.com/problems/combination-sum-ii/

题目描述:

给出一组候选数字(C)和目标数字(T),找出C中所有的组合,使组合中数字的和为T。C中每个数字在每个组合中只能使用一次。

跟39.Combination Sum差不多。注意解集不能包含重复的组合,比如[1,1] 1
解集只能是[[1]],不能是[[1],[1]]

思路:

排序再回溯。只不过多了一个对候选元素中重复元素的处理。同一元素不能重复使用。
使用prev参数每次检查candidates[i] == prev的话就continue。

代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> cmb;
    int len;
    void combinationSum2(vector<int>& candidates,int k,int curSum,int target){
        if(curSum>target){
            return;
        }
        else if(curSum==target){
            res.push_back(cmb);
        }
        else{
            int pre=-1;
            for(int i=k;i<len;i++){
                if(pre!=-1&& pre==candidates[i]){
                    continue;
                }
                pre=candidates[i];
                cmb.push_back(candidates[i]);
                combinationSum2(candidates,i+1,curSum+candidates[i],target);
                cmb.pop_back();
           }
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        len=candidates.size();
        if(len==0){
            return res;
        }
        sort(candidates.begin(),candidates.end());
        combinationSum2(candidates,0,0,target);
        return res;
    }
};

你可能感兴趣的:(LeetCode,回溯)