回溯算法理论基础
回溯算法的三部:
第一步:确定返回值和参数;
回溯算法中函数返回值一般为void;回溯算法参数可能有很多,根据具体题目再确定。
void backtracking(参数)
第二步:确定回溯算法终止条件;
if (终止条件) {
存放结果;
return;
}
第三步:确定回溯搜索的遍历过程。
for循环就是遍历集合区间,可以理解一个节点有多少个孩子,这个for循环就执行多少次。
backtracking这里自己调用自己,实现递归。
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
回溯函数模板:
最后确定回溯函数模板
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
思路:每个数组都要和其他数字组合,并且不重复;每取一个数就要缩小选择的范围,这里采用回溯处理。
解决:确定回溯三部曲
第一步:确定参数;
这里首先有两个参数,一是n相当于树的宽度,二是k相当于树的深度;其次还需要一个参数记录每次遍历的起点,1能和2,3,4;2能和3,4;3能和4;4最后没有,所以每次起点都要加1。
第二步:确定终止条件;
图中最后数字组合的数量是二,所以结束的时候就是每次组合的数组大小等于k的时候;这里需要定义一个存储每次组合结果的数组,vector<int> p。
第三步:确定遍历过程;
循环起点就是第一步中函数定义的起点参数,终止条件就是小于等于n,每次加1,从1遍历到4;循环内容应该分三步:①当前数字加入每次组合结果数组;②更新起点,递归剩余数组;③如果到达终止条件,最后加入数组的数字要出数组,例如数组中已经有[1,2],满足k=2,那么2就要出数组,再从3开始遍历。
最后需要一个记录全部结果的二维数组vector<vector<int>> r,将每次组合结果的数组,vector<int> p加入。
代码:
class Solution {
private:
vector<vector<int>> r;//存储所有的组合
vector<int> p;//储存当前整数符合条件的单一结果
void backtracking(int n,int k,int start){
if(p.size()==k){//终止条件
r.push_back(p);
return;
}
for(int i=start;i<=n;i++){
p.push_back(i);
backtracking(n,k,i+1);
p.pop_back();//已经满足条件的数字,需要去除,相当于回溯
}
}
public:
vector<vector<int>> combine(int n, int k) {
r.clear();
p.clear();
backtracking(n,k,1);
return r;
}
};
收获:
一、回溯法,一般可以解决如下几种问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
二、回溯理解
回溯法解决的问题都可以抽象为树形结构
因为回溯法解决的都是在集合中递归查找子集,集合的大小就构成了树的宽度,递归的深度,都构成的树的深度。