重拾C++之菜鸟刷算法第12篇---回溯算法

组合总和

题目

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。

对于给定的输入，保证和为 target 的不同组合数少于 150 个。

39. 组合总和 - 力扣（LeetCode）

题解

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& candidates, int target, int index){
        if(accumulate(path.begin(), path.end(), 0) == target){
            result.push_back(path);
            return;
        }
        if(accumulate(path.begin(), path.end(), 0) > target) return;

        for(int i = index; i < candidates.size(); i++){
            path.push_back(candidates[i]);
            backtracking(candidates, target, i);
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        backtracking(candidates, target, 0);
        return result;
    }
};

组合总和II

题目

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

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

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

40. 组合总和 II - 力扣（LeetCode）

题解

class Solution {
public:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int>& candidates, int target, int index, vector<bool> used){
        if(path.size() <= candidates.size()){
            int sum = accumulate(path.begin(), path.end(), 0);
            if(sum > target) return;
            if(sum == target){
                result.push_back(path);
                return;
            }
        }

        for(int i = index; i < candidates.size(); i++){
            if(i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) continue;
            path.push_back(candidates[i]);
            used[i] = true;
            backtracking(candidates, target, i + 1, used);
            used[i] = false;
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<bool> used(candidates.size(), false);
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, used);
        return result;
    }
};