题目:
题解:
class Solution:
def combinationSum3(self, k: int, n: int) -> List[List[int]]:
"""
回溯法,对于当前k和n, 枚举元素
"""
def backtracking(k: int, n: int, ans: List[int]):
if k == 0 or n <= 0:
# k == 0 表示已经使用了k个数,n <= 0表示这k个数元素和已经大于等于n,这些情况无需再枚举元素了
if k == 0 and n == 0:
# 使用k个数且恰好等于n,加入结果res
res.append(ans.copy())
return
# 枚举起点,等于当前数组中最后一个元素+1,数组为空相当于起始为1
start = (0 if len(ans) == 0 else ans[-1]) + 1
for i in range(start, 10):
if n - i < 0: # 加上元素i已经超过n,更大的i也就无须枚举了
break
ans.append(i) # 加入枚举的数
backtracking(k - 1, n - i, ans) # 递归搜索,k-1表示使用了一个数,n-i表示使用了i元素
ans.pop() # 弹出加入的数,还原现场
if n < (k + 1) * k // 2 or n > (19- k) * k // 2: return [] # n如果小于k能构成的最小值或者大于k能构成的最大值,直接返回
res = [] # 结果数组
ans = [] # 存储每一种组合
backtracking(k, n, ans) # 回溯法
return res