Leetcode216 组合总和III

最新推荐文章于 2024-07-04 21:39:57 发布

Jelly_Believer

最新推荐文章于 2024-07-04 21:39:57 发布

阅读量156

点赞数

分类专栏： LeetCode 回溯文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_42438346/article/details/115563407

版权

LeetCode 同时被 2 个专栏收录

67 篇文章 0 订阅

订阅专栏

回溯

15 篇文章 0 订阅

订阅专栏

@Leetcode
在这里插入图片描述相对于组合问题，只是多了一个限制，即所求集合总和为n；

回溯三步曲

确定递归函数参数

void generateSum(int k, int n, int sum, int startIndex,
vector &p)

确定终止条件

if(p.size() == k){
	if(sum == n)
       res.push_back(p);  //如果相等，则入数组
    return;//否则返回
}

单层搜索过程

for(int i = startIndex; i <= 9 - (k - p.size()) + 1; i++){
           if(sum + i <= n){
               sum = sum + i;
               p.push_back(i);

               generateSum(k, n, sum, i+1, p);

               sum -= i;
               p.pop_back();
           }

}

全部代码

class Solution {
private:
    vector<vector<int>> res;
    void generateSum(int k, int n, int sum, int startIndex, vector<int> &p){
        if(p.size() == k){
            if(sum == n)
                res.push_back(p);

            //for(int i = 0; i < p.size(); i++)
                //cout<<p[i];

            //cout<<endl;

            return;
        }

        for(int i = startIndex; i <= 9 - (k - p.size()) + 1; i++){
            if(sum + i <= n){
                sum = sum + i;
                p.push_back(i);

                generateSum(k, n, sum, i+1, p);

                sum -= i;
                p.pop_back();
            }

        }

        return;

    }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        int sum = 0;
        vector<int> p;
        generateSum(k, n, sum, 1, p);


        return res;

    }
};