代码随想录训练营Day24 | Leetcode 77
一、回溯算法理论基础
1、回溯法的定义
回溯法,即回溯搜索法,是一种搜索的方式。
回溯与递归是相辅相成的,只要有递归一定就有回溯。
回溯的本质:穷举,某些情况下可采取剪枝等优化操作。
2、回溯法解决的问题
组合问题:N个数按一定规则寻找k个数的集合;(不强调元素顺序,即无序)
切割问题:一个字符串按一定规则有几种切割方式;
子集问题:一个N个数的集合里有多少符合条件的子集;
排列问题:N个数按一定规则全排列,有几种排列方式;(强调元素顺序,即有序)
棋盘问题:N皇后,解数独。
注意:回溯法解决的问题:都可以抽象为树形结构,因为回溯法都是解决在集合中递归查找子集,集合的大小即树的宽度,递归的深度即树的深度。
递归需要终止条件,因此必然是一颗高度有限的树。
3、回溯法模板
第一,返回值和参数:通常返回值为void,参数则要由回溯逻辑确定;
第二,终止条件:通常是到达树的叶子节点结束本层递归,并return;
第三,回溯搜索的遍历过程:
void backtracking(参数)
{
if(终止条件)
{
存放结果;
return;
}
for(选择:本层集合元素)
{
处理节点;
backtracking(参数); //递归,注意参数中递归的体现
回溯,撤销处理结果;
}
}
二、77 组合
题目链接:77 组合
核心:明确终止条件是组合大小与k相同,以及参数中需要start指示当前遍历到n个数的哪一个,比如:第一次start从1开始,那么第二次则需要从2开始;另外,递归处理时要对start进行处理,比如:若start从2开始,那么递归搜索时就需要从3开始。
vector<int> path; //记录某个组合
vector<vector<int>> res; //记录所有组合的集合
void backtracking(int n,int k,int start)
{//回溯函数
//1、确定终止条件
if(path.size()==k)
{//组合大小为k,即满足组合条件,可以push到集合
res.push_back(path);
return;
}
//2、确定回溯遍历的搜索逻辑
for(int i=start;i<=n-(k-path.size())+1;++i)
{//循环时遍历i可进行优化,i<=n-(k-path.size())+1 (原:i<=n)
path.push_back(i); //处理节点
backtracking(n,k,i+1); //递归,即进入更深一层遍历,其实位置需+1
path.pop_back(); //回溯,撤销处理节点
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n,k,1); //调用回溯函数
return res;
}