刷题-Leetcode-40. 组合总和 II(回溯-组合-去重)

40. 组合总和 II

题目链接

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

题目描述

给定一个数组 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]
]

题目分析

本题candidates数组内有重复的，要在搜索的过程中就去掉重复组合。同一树枝上两个重复的元素可以同时取，同一树层上两个重复的元素不可以重复选取。去重的是同一树层上的“使用过”，树层去重的话，需要对数组排序！
使用used去重

class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(vector<int>& candidates, int target, int startIndex, int sum, vector<bool>& used){
        if(sum > target) return;
        if(sum == target){
            res.push_back(path);
            return;
        }
        for(int i = startIndex; i < candidates.size(); i++){
            // used[i - 1] == true，说明同一树支candidates[i - 1]使用过
            // used[i - 1] == false，说明同一树层candidates[i - 1]使用过
            if(i >= 1 && candidates[i] == candidates[i - 1] && used[i - 1] == false){//使用过
                continue;
            }
            path.push_back(candidates[i]);
            sum += candidates[i];
            used[i] = true;
            backtracking(candidates, target, i + 1, sum, used);//只能用一次 i+1
            sum -= candidates[i];
            used[i] = false;
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<bool> used(candidates.size(), false);
        res.clear();
        path.clear();
        // 首先把给candidates排序，让其相同的元素都挨在一起。
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, 0, used);
        return res;
    }
};

使用startIndex去重，更好理解。
注意i > startIndex 每次for循环，对第一个startIndex 之后的数字判定是否和前一个重复。

class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(vector<int>& candidates, int target, int startIndex, int sum){
        if(sum > target) return;
        if(sum == target){
            res.push_back(path);
            return;
        }
        for(int i = startIndex; i < candidates.size(); i++){
            // 要对同一树层使用过的元素进行跳过
            if(i > startIndex && candidates[i] == candidates[i - 1]){//注意这里 i>startIndex
                continue;
            }
            path.push_back(candidates[i]);
            sum += candidates[i];
            backtracking(candidates, target, i + 1, sum);//只能用一次 i+1
            sum -= candidates[i];
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        res.clear();
        path.clear();
        // 首先把给candidates排序，让其相同的元素都挨在一起。
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, 0);
        return res;
    }
};

使用set去重

class Solution {
private:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(vector<int>& candidates, int target, int startIndex, int sum){
        if(sum > target) return;
        if(sum == target){
            res.push_back(path);
            return;
        }
        unordered_set<int> uset;
        for(int i = startIndex; i < candidates.size(); i++){
            // 要对同一树层使用过的元素进行跳过
            if(uset.find(candidates[i]) != uset.end()){
                continue;
            }
            uset.insert(candidates[i]);
            path.push_back(candidates[i]);
            sum += candidates[i];
            backtracking(candidates, target, i + 1, sum);//只能用一次 i+1
            sum -= candidates[i];
            path.pop_back();
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        res.clear();
        path.clear();
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, 0);
        return res;
    }
};