【笔记】力扣 40. 组合总和 II——回溯算法中的去重

40. 组合总和 II

中等

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

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

注意：解集不能包含重复的组合。

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]

提示:

• 1 <= candidates.length <= 100
• 1 <= candidates[i] <= 50
• 1 <= target <= 30

题解

class Solution {
public:
    vector<int> cur;
    vector<vector<int>> res;
    void dfs(int start, int sum, vector<int>& candidates, int target) {
        if (sum>=target) {
            if (sum==target) {
                res.push_back(cur);
            }
            return;
        }
        if (start==candidates.size()) return;
        for (int i = start; i < candidates.size(); i++) {
            if (i > start && candidates[i] == candidates[i - 1]) continue; // 树层去重
            // 例如：1 1 4 9，题目所给数字里有重复的，但是只能使用一次，如果不去重会出现[1(第1个)49,1(第2个)49]这样重复的情况，因此需要去重；
            // 去重主要分为树层去重（dfs中的for）和树枝去重（dfs回溯）（见代码随想录https://www.bilibili.com/video/BV12V4y1V73A）
            // 树层去重的具体方法：从startIndex+1开始，如果与左侧元素相等，则跳过此数考虑往后选
            cur.push_back(candidates[i]);
            dfs(i + 1, sum + candidates[i], candidates, target);
            cur.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end()); // 排序，方便后面树层去重
        dfs(0,0,candidates,target);
        return res;
    }
};

若利用set或者把cur与res里的每个vector逐个数字比对会超时，而利用此方法可以在dfs中高效去重