代码随想录算法营DAY24 | 回溯算法理论基础 77. 组合 216.组合总和III 17. 电话号码的字母组合
回溯算法理论基础
文章链接
总结
1.回溯法解决的问题
- 回溯法,一般可以解决如下几种问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等等
2.如何理解回溯法
回溯法解决的问题都可以抽象为树形结构,是的,我指的是所有回溯法的问题都可以抽象为树形结构!
因为回溯法解决的都是在集合中递归查找子集,集合的大小就构成了树的宽度,递归的深度,都构成的树的深度。
递归就要有终止条件,所以必然是一棵高度有限的树(N叉树)。
回溯代码模板
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77. 组合
题目链接: 77. 组合
思路:
1.初始化两个数组,一个用来存放解的集合,一个用来存放单个解。
2.定义回溯函数:
(1)终止条件:当解的数组长度等于k的时候,存储解并终止。
(2)for循环:每次从上一层选中元素的下一个数开始,先把树放入解集cur,然后调用回溯函数,调用以后cur.pop()把这个数弹出,继续在本层for循环放入本层的其它数。
剪枝操作:
如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了,那么就没有必要搜索了。
接下来看一下优化过程如下:
- 已经选择的元素个数:path.size();
- 还需要的元素个数为: k - path.size();
- 在集合n中至多从该起始位置 : n - (k - path.size()) + 1,开始遍历
- 为什么有个+1呢,因为包括起始位置,我们要是一个左闭的集合。
所以优化之后的for循环是:
for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) // i为本次搜索的起始位置
代码:
//基础代码
var combine = function (n, k) {
let result=[];
let cur=[];
let start=1
let backtracking = function (n,k,start){
if (cur.length === k) {
result.push([...cur]);//注意这里浅拷贝和深拷贝
return;
}
for (let i = start; i <=n; i++) {
cur.push(i);
backtracking(n,k,i+1);//注意这里是i+1而不是start+1
cur.pop();
}
};
backtracking(n,k,start);
return result;
}
//剪枝操作
问题:
1.数组存进另外一个数组的时候,要用到深拷贝,否则会为空
2.调用回溯的时候,start下标更新的是i+1,而不是start+1,因为i会随着第一次选择的数字变化,否则就会出现这种错误:
216.组合总和III
题目链接: 216.组合总和III
思路:
代码:
var combinationSum3 = function(k, n) {
let res = [];
let path = [];
let sum = 0;
const dfs = (path,index) => {
// 剪枝操作
if (sum > n){
return
}
if (path.length == k) {
if(sum == n){
res.push([...path]);
return
}
}
for (let i = index; i <= 9 - (k-path.length) + 1;i++) {//注意这里也可以剪枝
path.push(i);
sum = sum + i;
index += 1;
dfs(path,index);
sum -= i
path.pop()
}
}
dfs(path,1);
return res
};
问题:
气死我了明明思路一模一样怎么代码就是过不了,没看出来哪错了。
17. 电话号码的字母组合
题目链接: 17. 电话号码的字母组合
思路:
代码:
var letterCombinations = function(digits) {
const k = digits.length;
const map = ["","","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"];
if(!k) return [];
if(k === 1) return map[digits].split("");
const res = [], path = [];
backtracking(digits, k, 0);
return res;
function backtracking(n, k, a) {
if(path.length === k) {
res.push(path.join(""));
return;
}
for(const v of map[n[a]]) {
path.push(v);
backtracking(n, k, a + 1);
path.pop();
}
}
};
问题:
思路懂了,代码不是自己写的,需要再写一遍。
感想总结
回溯还是有点难度啊…细节容易出错,需要自己多写。