[Leetcode] Combination Sum III

最新推荐文章于 2019-05-07 10:37:18 发布

Eathen_Zheng

最新推荐文章于 2019-05-07 10:37:18 发布

阅读量232

点赞数

分类专栏： Leetcode Leetcode 题解

本文链接：https://blog.csdn.net/eathen_zheng/article/details/54171227

版权

Leetcode 同时被 2 个专栏收录

73 篇文章 0 订阅

订阅专栏

Leetcode 题解

73 篇文章 0 订阅

订阅专栏

描述

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.

Example 1:

Input: k = 3, n = 7

Output:

[[1,2,4]]

Example 2:

Input: k = 3, n = 9

Output:

[[1,2,6], [1,3,5], [2,3,4]]

求所有由 1-9 这些数字构成的集合，使得集合里的元素个数为 $k$ ，元素的和为 $n$ 。

分析

这道题跟 Combination Sum 以及 Combination Sum II 的思路是一样的，对于这种求所有组合的题目优先想到用递归的方法解决。

因为知道元素可能的取值只有 $1$ 到 $9$ 这 $9$ 个数字，那么我们只需要依次考虑这几个数字，对于每个数字有“放入集合”以及“不放入集合”这两种情况，而每种情况下又是一个相同的问题，因此可以递归进行求解，直到集合里元素的个数等于 $k$ ，或者元素的和大于等于 n <script type="math/tex" id="MathJax-Element-320">n</script> 时，停止递归。

代码

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> res; vector<int> out;
        combine(k, n, 1, out, res);
        return res;
    }

    void combine(int k, int n, int num, vector<int>& out, vector<vector<int>>& res) {
        if (k == 0 && n == 0) {res.push_back(out); return;}
        if (num > 9 || n < 0 || k < 0) return;
        combine (k, n, num + 1, out, res);
        out.push_back(num);
        combine (k - 1, n - num, num + 1, out, res);
        out.pop_back();
    }
};