什么是回溯
回溯法也可以叫做回溯搜索法,它是一种搜索的方式。
回溯解决什么问题
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
回溯问题如何理解
回溯问题可以抽象成树结构,集合大小构成树的宽度,回溯深度构成树的深度。
说到这里,对于回溯问题还是一脸蒙蔽,所以回溯咋写呢??先别急,上模板!
回溯模板
void dfs(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
dfs(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
这个模板什么意思呢,if()中存放的是终止条件,for()中放的是每层树的结点,进行结点的处理,递归进入下一层,最后将处理的结点撤销(回溯前+x,回溯后自然就-x)
单看模板好像还是不明白,下面上题目!
力扣题目连接—组合
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例: 输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
怎么做呢,套模板!
void dfs(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
下面是leetcode的答题模板,思考如何往里套模板呢?
class Solution {
public List<List<Integer>> combine(int n, int k) {
}
}
首先,将模板套进去
class Solution {
public List<List<Integer>> combine(int n, int k) {
}
public void dfs(){
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
dfs(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
}
那么,终止条件是什么呢??
看题目——给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
返回k个数的组合,也就是说当列表或数组大小为k的时候终止
对于元素的选择呢??
题目也说了,返回范围[1,n],也就是说,for()里面是1-n,元素处理是怎样的呢,题目只是要求返回组合,那么很明显,对于元素,仅仅是放入数组或列表中。那么好像就有些思路了。
class Solution {
List<List<Integer>> res = new ArrayList();//返回的结果集
List<Integer> path = new ArrayList();//结果集中一个个小结果集
public List<List<Integer>> combine(int n, int k) {
dfs(n,k,1);
return res;
}
public void dfs(int n,int k,int index){
if(path.size() == k){
res.add(new ArrayList(path));
return ;
}
for(int i = index;i <= n;i++){//index为从哪个数开始,从index开始一直到n
path.add(i);
dfs(n,k,i+1);
path.remove(path.size()-1);
}
}
}
咦,好像对于回溯有点思路了,那接下来继续上题目巩固!
力扣题目——子集
题目描述:给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
老样子,套模板!
依旧是先分析,什么时候递归终止。
从第一个元素开始,当索引大于等于数组下标最大值的时候,递归终止。
for里面放什么?明显,从索引位置开始,一直到数组最后。所以好像有些明白怎么写了
class Solution {
List<List<Integer>> res = new ArrayList();
List<Integer> path = new ArrayList();
public List<List<Integer>> subsets(int[] nums) {
dfs(nums,0);
return res;
}
public void dfs(int[] nums,int index){
res.add(new ArrayList(path));
if(index >= nums.length){
return ;
}
for(int i = index;i < nums.length;i++){
path.add(nums[i]);
dfs(nums,i+1);
path.remove(path.size() - 1);
}
}
}
今日回溯就先到这里。当然,这些仅仅是回溯的冰山一角,要想真正掌握回溯,刷题是必不可少的,推荐大家去leetcode官网去多刷题。此外,在本文中只是写了回溯最基本的写法,并没有考虑减枝优化,有兴趣的小伙伴可以去想想如何减枝优化。