LeetCode：40 组合总数 II 去重剪枝dfs

题目描述

给定一个数组 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

这题去重的思路和【LeetCode：47 全排列II】相似，但是也有值得注意的地方

如图重复情况出现如下：

如图 [1,2,5] 和 [1', 2,5]是重复的，但是 [1, 1', 6]又是合法的

我们对数组排序（使得相同数字挨在一起），然后人为的对相同数字进行区分，假设一个序列有 a 和 a’

总是认为 a 必须出现在 a‘ 之前

在每次选取之前，观察前面的数字是否相同（即出现 a a’ 的情况），如果相同并且前面的数字 a 未被选择，那么说明我们不能选择 a’，因为人为规定 a 必须出现在 a’ 之前

if(i>=1 && candidates[i]==candidates[i-1] && !vis[i-1]) continue;

(或者说这样做排除了 “同一个数字在同样的位置出现第二次” 的情况)

代码

class Solution {
public:
    vector<vector<int>> ans;
    void dfs(int n, vector<int> &vec, vector<bool> &vis, vector<int> &candidates, int start)
    {
        if(n<0) return;
        if(n==0) {ans.push_back(vec); return;}
        for(int i=start; i<candidates.size(); i++)
        {
            if(i>=1 && candidates[i]==candidates[i-1] && !vis[i-1]) continue;
            if(!vis[i])
            {
                vis[i] = true;
                vec.push_back(candidates[i]);
                dfs(n-candidates[i], vec, vis, candidates, i+1);
                vec.pop_back();
                vis[i] = false;
            }
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target)
    {
        sort(candidates.begin(), candidates.end());
        vector<bool> vis(candidates.size(), false);
        vector<int> vec;
        dfs(target, vec, vis, candidates, 0);
        return ans;
    }
};

或者更简单的
当我们在位置x选择一个数a的时候，如果在枚举x位置的其他选择的时候第二次遇到a，那么直接跳过，那么也不需要visited数组了

注意我们排除的是同样的数字在同样的位置出现第二次的情况

if(i>start && candidates[i]==candidates[i-1]) continue;

代码

class Solution {
public:
    vector<vector<int>> ans;
    void dfs(int n, vector<int> &vec, vector<int> &candidates, int start)
    {
        if(n<0) return;
        if(n==0) {ans.push_back(vec); return;}
        for(int i=start; i<candidates.size(); i++)
        {
            if(i>start && candidates[i]==candidates[i-1]) continue;
            vec.push_back(candidates[i]);
            dfs(n-candidates[i], vec, candidates, i+1);
            vec.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target)
    {
        sort(candidates.begin(), candidates.end());
        vector<int> vec;
        dfs(target, vec, candidates, 0);
        return ans;
    }
};