python 组合求和_组合求和III（Python实现）,每日,水题,216,总和,IIIPython

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