Leetcode 216.组合总和III

Posted on |

216.组合总和III

题目

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

说明：

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

1
2
3
4
5
6
7
8
示例 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(object):
    def combinationSum3(self, k, n):
        """
        :type k: int
        :type n: int
        :rtype: List[List[int]]
        """
        res=[]
        self.backtrack(k, n, res, [], 1)
        return res

    def backtrack(self, k, target, res, temp, start):
        if len(temp)>k:
            return
        elif len(temp)==k and target==0:
            res.append(list(temp))
        else:
            for i in range(start,10):
                temp.append(i)
                self.backtrack(k, target-i, res, temp, i+1)
                temp.pop()