leetcode数组刷题(五)——回溯组合

leetcode数组刷题(五)——回溯组合

39、组合总和
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
递归回溯什么的目前还不是很会,多找几个学习一下

class Solution {
public:
    void dfs(vector<int>& candidates, int target, vector<vector<int>>& ans, vector<int>& combine, int idx) {
        if (idx == candidates.size()) {
            return;
        }
        if (target == 0) {
            ans.push_back(combine);
            return;
        }
        // 左分支
        dfs(candidates, target, ans, combine, idx + 1);
        // 选择当前数右分支
        if (target - candidates[idx] >= 0) {
            combine.push_back(candidates[idx]);
            dfs(candidates, target - candidates[idx], ans, combine, idx);
            combine.pop_back();//不是很理解,可能惯常操作
        }
    }

    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        vector<int> combine;
        dfs(candidates, target, ans, combine, 0);
        return ans;
    }
};

40、组合总和 II
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
注意:解集不能包含重复的组合。

class Solution {
public:
    void dfs(vector<int>& candidates, int target, vector<vector<int>>& ans, vector<int>& combine, int idx) {
        if (target == 0) {
            ans.push_back(combine);
            return;
        }
        if (idx == candidates.size()) return;
        // 无右分支,不可选同一个数字
        //dfs(candidates, target, ans, combine, idx + 1);
        // 左分支
        for (int i=idx;i<candidates.size() && target - candidates[idx] >= 0;i++) {
            if (i>idx && candidates[i]==candidates[i-1]) continue;//判重模板
            combine.push_back(candidates[i]);
            dfs(candidates, target - candidates[i], ans, combine, i+1);
            combine.pop_back();//不是很理解,可能惯常操作
        }
    }

    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<vector<int>> ans;
        vector<int> combine;
        sort(candidates.begin(),candidates.end());
        dfs(candidates, target, ans, combine, 0);
        return ans;
    }
};

感觉用循环之后的理解难度比上一题的官方答案简单,就是其实都是对选定一个元素后,对其他元素都进行迭代即可,上一题还分为左右,没有必要,上一题的代码写成循环如下,感觉区别就是元素不能重复选择,则迭代进去的索引从下一个元素开始,同时需要判断答案的重复,就是之前一直用的判重模板:

class Solution {
public:
    void dfs(vector<vector<int>>& res, vector<int>& path, vector<int>& candidates, int target, int sum, int start)
    {
        if (sum > target) return;
        if (sum == target)
        {
            res.push_back(path);
            return;
        }
        for (int i = start; i < candidates.size(); ++i)
        {
            path.push_back(candidates[i]);
            dfs(res, path, candidates, target, sum + candidates[i], i);
            path.pop_back();
        }
    }
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<vector<int>> res;
        vector<int> path;
        dfs(res, path, candidates, target, 0, 0);
        return res;
    }
};

你可能感兴趣的:(leetcode,leetcode,深度优先,算法)