Python | Leetcode Python题解之第216题组合总和III

题目：

题解：

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