【leetcode】40 组合总和II（回溯）

题目链接：https://leetcode-cn.com/problems/combination-sum-ii/

题目描述

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明：

所有数字（包括目标数）都是正整数。
解集不能包含重复的组合。
示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combination-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

1 回溯-set去重

与题39不同之处在于（1）candidates的数字有重复（2）
candidates 中的每个数字在每个组合中只能使用一次

所以我们考虑用set来存储已经出现的组合，同时由于candidates的数字有重复，为了避免重复元素构成的组合，所以需要先对数组进行排序。

以 candidates = [10,1,2,7,6,1,5], target = 8,为例：
如果不排序，即使用set存储，因为数组中有两个1，仍然会出现[1,7]和[7,1]这样的重复组合，所以需要进行排序使组合内部是从小到大排序的。

最后将set赋值到vector返回。

class Solution {
private:
    set<vector<int>> s; // set去重
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<int> curr;
        sort(candidates.begin(),candidates.end());  // 排序使数组内部也是从小到大排序
        combinationSumCore(candidates, target, 0, curr);
        vector<vector<int>> ret(s.begin(),s.end());
        return ret;
    }

    void combinationSumCore(vector<int>& candidates, int target, int begin, vector<int>& curr){
        if(target == 0){
            s.insert(curr);
            return;
        }
        else if(target<0||begin>= candidates.size())           // curr和大于target
            return;
        for (int i = begin; i < candidates.size(); ++i) {   // 注意用变量begin控制curr数组内的数字都是从小到大
            curr.push_back(candidates[i]);
            combinationSumCore(candidates, target-candidates[i], i+1, curr); // target-candidates[i]表示选择i元素
            curr.pop_back();
        }
    }
};

2 回溯-不用set

其实思路和上面是一样的，重点在设置条件去重

if(i>begin&&candidates[i-1]==candidates[i]) //目的是去重
                    continue;

确保了组合中，以某元素开头只能有一次；
以 candidates = [10,1,2,7,6,1,5], target = 8, 为例，不会出现两次[1,7]

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<int> vec;
        sort(candidates.begin(),candidates.end());  //排序
        dpSum(candidates,target,vec,0);
        return ans;
    }

private:
    vector<vector<int>> ans;
    void dpSum(vector<int>& candidates, int target,vector<int> vec,int begin){
        if(target<0)
            return;
        else if(target==0){
            ans.push_back(vec);
        }else{
            for(int i=begin;i<candidates.size()&&candidates[i]<=target;++i){
                //目的是去重
                if(i>begin&&candidates[i-1]==candidates[i])
                    continue;
                vec.push_back(candidates[i]);
                dpSum(candidates,target-candidates[i],vec,i+1);
                vec.pop_back();
            }
        }
    }
};