博客

解题思路

参考

组合总和



该解题思路来自代码随想录

如果

used[i-1]

为真，说明我还在取

i-1

这一支上的树，还没到同层

如果

used[i-1]

为假，说明

i-1

这一支已经结束了，现在开始同层的另一支

代码

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking (vector<int>& nums, vector<bool>& used) {
        if(path.size() == nums.size()) {
            result.push_back(path);
            return;
        }
        
        for(int i = 0;i < nums.size();++i) {
            //used[i-1]为真，证明我还在取这一支上的数，还没到同层
            if(i > 0 && nums[i] == nums[i-1] && !used[i-1])continue;
            //如果这个数被使用过就跳过，这样就可以每次都从0开始往后取数
            if(!used[i]){
                used[i] = 1;
                path.push_back(nums[i]);
                backtracking(nums,used);
                //这一支结束，进入同层的下一支
                path.pop_back();
                used[i] = 0;
            }
        }
    }
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
       sort(nums.begin(),nums.end());
       vector<bool> used(nums.size(),0);
       backtracking(nums,used);
       return result;
    }
};

用这样的去重方法改写组合总和：

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

    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        DFS(candidates, target, 0);
        return ans;
    }
    void DFS(vector<int> candidates, int target, int level) {
        if (target < 0)return;//这个可以不要，在for循环中有判断
        if (target == 0) {
            ans.push_back(path);
            return;
        }

        for (int i = level; i < candidates.size() && candidates[i] <= target; i++) {
            //如果进入了for循环，代表在同一层，同一层有用过直接跳过
            if(i > level && candidates[i] == candidates[i-1])continue;
            path.push_back(candidates[i]);
            DFS(candidates, target - candidates[i], i + 1);
            //进行回溯
            path.pop_back();
        }
        return;
    }
};

这里用level去重，和全排列不一样的是：全排列需要从头开始每个数都选到，而组合问题是从前往后的，所以可以直接用level去重。