LeetCode之Combination Sum III

最新推荐文章于 2020-09-12 22:04:03 发布

原创最新推荐文章于 2020-09-12 22:04:03 发布 · 312 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一个使用深度搜索算法解决组合总和III问题的方法，目标是从1到9的整数中找到所有可能的k个数字的组合，使得它们的和等于给定的目标值n。

/*深度搜索。采用深度搜索，每次添加一个数字。*/
class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int> > res;
        if(k <= 0 || n < (k+1)*k/2) return res;
        vector<int> path;
        for(int i = 1; i < 10; ++i) combinationSum3(res, path, i, n, k);
        return res;
    }
    
    void combinationSum3(vector<vector<int> > &res, vector<int> &path, int cur, int sum, int k){
        path.push_back(cur);
        sum -= cur;
        if(path.size() == k){
            if(sum == 0) res.push_back(path);
            path.pop_back();
            sum += cur;
            return;
        }
        for(int i = cur + 1; i < 10; ++i) combinationSum3(res, path, i, sum, k);
        path.pop_back();
        sum += cur;
    }
};