题目:
77.组合
学习内容:
回溯算法理论基础
回溯与递归是紧密结合的,有递归就会有回溯。其本质是穷举所有的可能,一般可以解决如下问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
注意:组合无序,排列有序。
回溯法解决的问题都是在集合中递归查找子集,可以抽象为树形结构。 回溯有三部曲:
- 回溯函数模板返回值以及参数
- 回溯函数终止条件
- 回溯搜索的遍历过程
其算法模板框架如下:
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(); // 回溯
}
}
}
剪枝操作
- 已经选择的元素个数:path.size();
- 还需要的元素个数为: k - path.size();
- 在集合n中最多要从该起始位置 : n - (k - path.size()) + 1,开始遍历
因此,我们要将部分代码改成
for (int i = startIndex; i <= n - (k - path.size()) + 1; i++)
学习时间:
2024.4.1