代码随想录算法训练营第二十四天 | 回溯算法part01

题目：

77.组合

---

学习内容：

回溯算法理论基础

回溯与递归是紧密结合的，有递归就会有回溯。其本质是穷举所有的可能，一般可以解决如下问题：

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

注意：组合无序，排列有序。
回溯法解决的问题都是在集合中递归查找子集，可以抽象为树形结构。 回溯有三部曲：

1. 回溯函数模板返回值以及参数
2. 回溯函数终止条件
3. 回溯搜索的遍历过程

其算法模板框架如下：

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

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

77.组合

这道题主要是套模板

class Solution {
    List<List<Integer>> result = new ArrayList<>(); // 保存最终的结果，二维数组
    List<Integer> path = new ArrayList<>(); // 保存每条path
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return result;
    }
    public void backtracking(int n, int k, int startIndex) { // 1. 确定回溯函数与参数
        // 2. 确定终止条件:到叶子节点(path大小等于k)，将path扔到result并终止
        if (path.size() == k) {
            result.add(new ArrayList<>(path)); // 注意这里要新建path，因为要保存多个path
            return;
        }
        // 3. 确定单层回溯,for是横向遍历，递归是纵向遍历,startIndex是确定递归的起点
        for (int i = startIndex; i <= n; i++) {
            path.add(i); // 处理节点，将i扔到path
            backtracking(n, k, i + 1); // 递归
            path.removeLast(); // 回溯
        }
    }
}

剪枝操作

1. 已经选择的元素个数：path.size();
2. 还需要的元素个数为: k - path.size();
3. 在集合n中最多要从该起始位置 : n - (k - path.size()) + 1，开始遍历
   因此，我们要将部分代码改成

for (int i = startIndex; i <= n - (k - path.size()) + 1; i++)

学习时间：

2024.4.1