leetcode [216] 组合总和 III

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

找出所有相加之和为 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]]

控制回溯深度为k，满足条件加入答案。

class Solution:
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        ans = []
        def core(height, cur, prev, prev_sum):
            if height == k:
                if sum(cur) == n:
                    ans.append(cur)
                else:
                    return
            for i in range(prev + 1, 10):
                core(height + 1, cur + [i], i, prev_sum + i)
        core(0, [], 0, 0)
        return ans