找出所有相加之和为 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、回溯
Python解答:
class Solution:
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
res = []
def dfs(n, k, i, tem):
if k == 0 and n == 0:
res.append(deepcopy(tem))
return
elif k > 0 and n == 0:
return
elif k == 0 or n < 0:
return
else:
for j in range(i, 10):
dfs(n-j, k-1, j+1, tem+[j])
dfs(n, k, 1, [])
return res
时间复杂度:O(C_n^k*k)
留言