Leetcode—216.组合总和III【中等】-Toy模板网

这篇具有很好参考价值的文章主要介绍了Leetcode—216.组合总和III【中等】。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

2023每日刷题（七十八）

Leetcode—216.组合总和III

Leetcode—216.组合总和III【中等】,LeetCode刷题,leetcode,深度优先,算法,dfs,剪枝,回溯,经验分享

算法思想

Leetcode—216.组合总和III【中等】,LeetCode刷题,leetcode,深度优先,算法,dfs,剪枝,回溯,经验分享

实现代码

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> ans;
        vector<int> path;
        function<void(int, int)> dfs = [&](int i, int res) {
            int d = k - path.size();
            if(res < 0 || res > (2 * i - d + 1) * d / 2) {
                return;
            }
            if(d == 0) {
                ans.emplace_back(path);
                return;
            }
            for(int j = i; j >= d; j--) {
                path.emplace_back(j);
                dfs(j - 1, res - j);
                path.pop_back();
            }
        };
        dfs(9, n);
        return ans;
    }
};

运行结果

Leetcode—216.组合总和III【中等】,LeetCode刷题,leetcode,深度优先,算法,dfs,剪枝,回溯,经验分享

选或不选实现代码

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> ans;
        vector<int> path;
        function<void(int, int)> dfs = [&](int i, int res) {
            int d = k - path.size();
            if(res < 0 || res > (2 * i - d + 1) * d / 2) {
                return;
            }
            if(d == 0) {
                ans.emplace_back(path);
                return;
            }
            // 不选
            if(i > d) {
                dfs(i - 1, res);
            }
            // 选
            path.emplace_back(i);
            dfs(i - 1, res - i);
            path.pop_back();
        };
        dfs(9, n);
        return ans;
    }
};