Leetcode—216.组合总和III【中等】

2023每日刷题(七十八)

Leetcode—216.组合总和III

Leetcode—216.组合总和III【中等】_第1张图片

算法思想

Leetcode—216.组合总和III【中等】_第2张图片
Leetcode—216.组合总和III【中等】_第3张图片

实现代码

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【中等】_第4张图片

选或不选实现代码

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;
    }
};

运行结果

Leetcode—216.组合总和III【中等】_第5张图片

之后我会持续更新,如果喜欢我的文章,请记得一键三连哦,点赞关注收藏,你的每一个赞每一份关注每一次收藏都将是我前进路上的无限动力 !!!↖(▔▽▔)↗感谢支持!

你可能感兴趣的:(LeetCode刷题,leetcode,深度优先,算法,dfs,剪枝,回溯,经验分享)