算法训练第二十四天 | 回溯算法 77.组合

回溯算法

可解决的问题

回溯法，一般可以解决如下几种问题：

• 组合问题：N个数里面按一定规则找出k个数的集合
• 切割问题：一个字符串按一定规则有几种切割方式
• 子集问题：一个N个数的集合里有多少符合条件的子集
• 排列问题：N个数按一定规则全排列，有几种排列方式
• 棋盘问题：N皇后，解数独等等

代码模板

思路就是通过横向for循环，纵向递归的形式（递归里面嵌for循环套）

代码模板如下

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (本层集合中元素（树中节点孩子的数量就是集合的大小）) {
        处理节点;
        backtracking(路径，选择的列表); // 递归
        回溯，撤销处理结果
    }
}

LeetCode 77.组合

题目链接

77.组合

思路

理解模板，找到迭代的终止条件，需要传入的参数，以及单次迭代的逻辑。result用来存最后的结果，path就是收集路径上的数据的意思，也就是叶子节点的内容（即一个组合的结果）

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []
        path = []
        def backtracking(n, k, startIndex):
            if len(path) == k:
                result.append(path[:]) //一定要写[:]
                return
            for i in range(startIndex, n+1):  //可以优化的地方
                path.append(i)
                backtracking(n, k, i+1)
                path.pop()
            
        backtracking(n, k, 1)
        return result

做剪枝优化，已经明确不可能会得到k个节点的分枝直接剪枝，也就是对for循环的遍历范围做调整

总结

回溯算法通俗解释就是一个递归算法内嵌套了一个for循环。在写代码的时候按照回溯的模板，写好递归三部曲的每个地方的内容。