【leetcode】216 组合总和III（回溯）

题目链接：https://leetcode-cn.com/problems/combination-sum-iii/

题目描述

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

说明：

• 所有数字都是正整数。
• 解集不能包含重复的组合。

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

代码

class Solution {
    vector<int> candidates = {1,2,3,4,5,6,7,8,9};
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        vector<vector<int>> ret;
        vector<int> curr;
        if(n<1 || k < 1 || n>(1+9)*9/2 || k > 9) return ret;    // 小于0或超过表示范围
        combinationSum3Core(k,n,0,curr,ret);
        return ret;
    }
    void combinationSum3Core(int k, int n, int begin, vector<int>& curr, vector<vector<int>>& ret){
        if(n < 0) return;
        if(n == 0){
            if(curr.size() == k)            // 和为n并且个数为k
                ret.push_back(curr);
            return;
        } else{
            if(curr.size() >= k) return;    // 组合数字个数超过k
            for (int i = begin; i < candidates.size(); ++i) {
                curr.push_back(candidates[i]);
                combinationSum3Core(k,n-candidates[i], i+1, curr, ret);
                curr.pop_back();
            }
        }
    }
};