代码随想录算法训练营第二十四天 | 回溯算法理论基础与 leetcode 77.组合

回溯算法理论基础

前情回顾： 递归里有回溯

回溯算法的本质：穷举所有的可能， 把回溯算法看成一棵树

回溯三部曲：
1. 确定回溯函数的返回值及其参数（通常为返回void）
2. 算法的终止条件
3. 算法的遍历过程，for循环横向遍历各集合的大小，递归即纵向遍历所有子集合

77.组合

题目：给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

题目链接：77. 组合

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        result = [] # 全部
        path = [] # 单次符合结果的返回值

        def backtracking(n, k, startid):
            if len(path) == k:
                result.append(path[:])
                return

            # 剪枝优化
            # 如果for循环选择的起始位置之后的元素个数已经不足需要的元素个数了，那么就没有必要搜索了
            last_startidx = n - (k - len(path)) + 1 # 还需要选择的元素个数，左闭区间
            for x in range(startid, last_startidx + 1):
                path.append(x) # 添加结点
                backtracking(n, k, x + 1) # 递归
                path.pop() # 回溯,  撤销处理的结点，继续尝试穷举当前情况下符合条件的组合


        backtracking(n, k, 1)
        return result