Leetcode 216. 组合总和 III(Python3)

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]]

---

组合总和的题都用递归，也都可以改迭代

class Solution:
    def combinationSum3(self, k, n):
        """
        :type k: int
        :type n: int
        :rtype: List[List[int]]
        """
        if not k or k>9:
            return []
        self.res=[]
        self.dfs([],k,n)
        return self.res
        
    def dfs(self,cr,k,n):  #cr = current_result
        if not k and not n:
            self.res.append(cr)
            return
        if n < 1 or k < 1:
            return
        start = cr[-1] if cr else 0
        for i in range(start+1,10):
            self.dfs(cr+[i],k-1,n-i)

---

大佬一步到位，Python式解法

class Solution:
    def combinationSum3(self, k, n):
        from itertools import combinations
        return [ item for item in list(combinations(range(1, 10), k)) if sum(item) == n]