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

2023每日刷题（七十八）

Leetcode—216.组合总和III

在这里插入图片描述

算法思想

在这里插入图片描述

实现代码

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

运行结果

在这里插入图片描述

选或不选实现代码

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