代码随想录算法训练营第二十四天|回溯算法理论基础、77. 组合
回溯算法理论基础
题目:回溯算法理论基础
文章讲解:https://programmercarl.com/%E5%9B%9E%E6%BA%AF%E7%AE%97%E6%B3%95%E7%90%86%E8%AE%BA%E5%9F%BA%E7%A1%80.html
视频讲解:https://www.bilibili.com/video/BV1cy4y167mM/
状态:
看完代码随想录之后的想法
回溯和递归相辅相成,通常都在递归函数的下面。
回溯法能解决哪些问题:
- 组合问题:eg 给你一个集合 1,2,3,4;找出包含2的两两组合(无顺序)
- 切割问题:eg 给你一个字符串,如何切割保证字符串为回文串
- 子集问题:eg 给你1,2,3,4 求出所有的组合子集
- 排列问题:eg 1,2;2,1两种排列,有顺序
- 棋盘问题:n皇后、解数独
如何理解回溯法
将回溯法抽象成图形结构。
回溯算法模版
void backtracking(参数){
if(终止条件){
收集结果
return;
}
// 进入单层搜索逻辑,一般情况下是一个for循环
for(集合元素集){
处理节点
递归函数
回溯操作(撤销的过程,如1,2处理完后弹出2,处理1,3)
}
}
组合
77. 组合
文章讲解:https://programmercarl.com/0077.%E7%BB%84%E5%90%88.html#%E6%80%9D%E8%B7%AF
题目链接:https://leetcode.cn/problems/combinations/
视频讲解:https://www.bilibili.com/video/BV1ti4y1L7cv/
自己看到题目的第一想法
直接按照模版套,没想出来该怎么写
看完代码随想录之后的想法
回溯法解决的问题都可以抽象为树形结构(N叉树),用树形结构来理解回溯就容易多了。
这棵树,一开始集合是 1,2,3,4, 从左向右取数,取过的数,不再重复取。
第一次取1,集合变为2,3,4 ,因为k为2,我们只需要再取一个数就可以了,分别取2,3,4,得到集合[1,2] [1,3] [1,4],以此类推。
每次从集合中选取元素,可选择的范围随着选择的进行而收缩,调整可选择的范围。
图中可以发现n相当于树的宽度,k相当于树的深度。
图中每次搜索到了叶子节点,我们就找到了一个结果。
只要把达到叶子节点的结果收集起来,就可以求的n个数中k个数的组合集合。
回溯三部曲:
- 递归函数的返回值以及参数
List<List<Integer>> result; // 存放符合条件结果的集合
List<Integer> path; // 用来存放符合条件结果
参数
void backtracking(int n, int k, int startIndex)
- 回溯函数终止条件
什么时候达到叶子节点呢?当path这个数组的大小达到了k,表明我们找到了一个子集大小为k的组合了,在图中path存的就是根节点到叶子节点的路径。
if (path.size() == k) {
result.add(path);
return;
}
- 单层循环逻辑
- for循环用来横向遍历,递归过程是纵向遍历。
for (int i = startIndex; i <= n; i++) { // 控制树的横向遍历
path.add(i); // 处理节点
backtracking(n, k, i + 1); // 递归:控制树的纵向遍历,注意下一层搜索要从i+1开始
path.remove(path.size() - 1); // 回溯,撤销处理的节点
}
- 剪枝的处理
如果n=4,k=4的话,第一层for循环从2开始的遍历都没意义了。第二层for循环,从元素3开始的遍历都没有意义了。如图:
优化过程:
- 已选择的元素个数:path.size();
- 还需要的元素个数为:k - path.size();
- 在集合n中至多要从该起始位置 : n - (k - path.size()) + 1,开始遍历
自己实现过程中遇到哪些困难
整体还是比较复杂,还需要熟悉回溯的公式。需要将题目抽象为树形结构。
今日收获&学习时长
回溯算法的核心就是将题目抽象为树形结构,按照回溯三步骤来写代码。
- 确定递归函数的参数和返回值
- 回溯终止条件
- 单层循环逻辑处理