(每日水题)216. 组合总和 III(Python实现)

找出所有相加之和为 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:
    # 从9个数中选择k-1个数,k<10 ,直接DFS
    def dfs(self, pos, s, last): # s为当前需要组合的和,last为当前需要数字的个数
        for i in range(pos+1,11-last):
            self.vis[i] = 1
            if last - 1 == 0 and s-i == 0:
                temp = []
                for i in range(1,i+1):
                    if self.vis[i]:
                        temp.append(i)
                self.ret.append(temp)
            if s - i <=0:
                self.vis[i]=0
                break;
            if last - 1 > 0:
                self.dfs(i, s-i, last-1)
            self.vis[i] = 0
    def combinationSum3(self, k: int, n: int) -> List[List[int]]:
        self.vis = [0] * 10 # 用于标志1到9哪个数被选取了
        self.ret = []
        self.dfs(0, n, k)
        return self.ret

 

你可能感兴趣的:(python,日常一水,算法,python,dfs,java,c++)