https://leetcode.cn/problems/combination-sum-iii/
找出所有相加之和为 n 的 k 个数的组合,且满足下列条件:
只使用数字1到9
每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。
class Solution:
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
# 星级:☆☆☆
# 标签:回溯
from copy import deepcopy
nums = [i for i in range(1, 10)]
path, result = [], []
def backtrack(start):
if sum(path) == n and len(path) == k:
result.append(deepcopy(path)) # 或者 result.append(path[:])
return
for i in range(start, 9):
path.append(nums[i])
backtrack(i + 1)
path.pop()
backtrack(0)
return result