Leetcode 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

代码1

用一个vis数组判断前面的数字是否在本层使用过

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;

    void backtracking(vector<int>& candidates, int target, int sum, int index, vector<bool>& vis) {
        if (sum > target) return; // 剪枝
        if (sum == target) {
            res.push_back(path);
            return;
        }
        // candidates[i] == candidates[i - 1] && vis[i - 1] == true
        // 代表本层中和现在这个数相同的数在同一个树枝中被选择过了，可以继续选择
        for (int i = index; i < candidates.size() && sum + candidates[i] <= target; i++) {
            if (i > 0 && candidates[i] == candidates[i - 1] && vis[i - 1] == false) { // 注意是 i > 0
                continue; // candidates[i] == candidates[i - 1] && vis[i - 1] == false
                          // 代表本层中和现在这个数相同的数已经被选择过了，继续选择就是重复 
            }
            sum += candidates[i];
            path.push_back(candidates[i]);
            vis[i] = true;
            backtracking(candidates, target, sum, i + 1, vis);
            sum -= candidates[i];
            path.pop_back();
            vis[i] = false;
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        res.clear();
        path.clear();
        if (candidates.size() == 1 && candidates[0] < target) return res;
        vector<bool> vis(candidates.size(), false); // 标记某个位置的数是否使用了
        sort(candidates.begin(), candidates.end());// 排序便于去重
        backtracking(candidates, target, 0, 0, vis);

        return res;
    }
};

代码2

用index来保证本层不选择相同的数字

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;

    void backtracking(vector<int>& candidates, int target, int sum, int index) {
        if (sum > target) return; // 剪枝
        if (sum == target) {
            res.push_back(path);
            return;
        }

        for (int i = index; i < candidates.size() && sum + candidates[i] <= target; i++) {
            if (i > index && candidates[i] == candidates[i - 1]) { // 注意是 i > index
                continue; // 保证同一层不选择一样的数字
            }
            sum += candidates[i];
            path.push_back(candidates[i]);
            backtracking(candidates, target, sum, i + 1);
            sum -= candidates[i];
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        res.clear();
        path.clear();
        if (candidates.size() == 1 && candidates[0] < target) return res;
        sort(candidates.begin(), candidates.end());// 排序便于去重
        backtracking(candidates, target, 0, 0);

        return res;
    }
};