题目:
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
所有数字都是正整数。
解集不能包含重复的组合。
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
思路:
又是一道组合,挺好的,书读百遍其义自见,同类型的题多做才能掌握窍门儿。
同样是按照之前的思路,其实就是传说中的回溯法了。
套路相同,不同的地方:递归终止条件
temp之和大于n,或者k0
k0的时候,有两种情况:
1,加入结果集
2,不加入结果集
from typing import List
class Solution:
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
src = [i for i in range(1,10)]
self.res = []
def dfs(k,l,n,temp):
if sum(temp)>n:
return
elif k == 0 and sum(temp) == n:
self.res.append(temp[:])
return
elif k == 0 and sum(temp) != n:
return
size = len(l)
for i in range(size):
temp.append(l[i])
dfs(k-1,l[i+1:],n,temp)
temp.pop()
dfs(k,src,n,[])
return self.res
if __name__ == '__main__':
k = 3
n = 15
res = Solution().combinationSum3(k=k,n=n)
print(res)