题目:
题解:
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,加入结果resres.append(ans.copy())return# 枚举起点,等于当前数组中最后一个元素+1,数组为空相当于起始为1start = (0 if len(ans) == 0 else ans[-1]) + 1for i in range(start, 10):if n - i < 0: # 加上元素i已经超过n,更大的i也就无须枚举了breakans.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