leetcode 216. 组合总和 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]]

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution:
    def combinationSum3(self, k, n):
        """
        :type k: int
        :type n: int
        :rtype: List[List[int]]
        """
        res = []
        def dfs(res,tmp,idx,sum1):
            if len(tmp) == k:
                if sum1 == n:
                    res += [tmp]
                else:
                    return
            if len(tmp) > k:return

            for i in range(idx,10):
                dfs(res,tmp+[i],i+1,sum1+i)
        
        dfs(res,[],1,0)
        return res