找出所有相加之和为 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]]
组合
全组合然后筛选出符合条件的结果。
Code
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
ans, nums = [], [1, 2, 3, 4, 5, 6, 7, 8, 9]
for item in itertools.combinations(nums, k):
if sum(item) == n:
ans.append(list(item))
return ans