5、回溯算法
最后更新于:2022-04-02 06:11:24
回溯算法(backtracking)是一个类似枚举的搜索尝试过程,在寻找问题解的过程中,当发现不满足求解条件时,就退回一步,尝试其它路径,该算法的时间复杂度都不会低于 O(N!),复杂的例题包括[正则表达式匹配](https://leetcode-cn.com/problems/regular-expression-matching/)、[解数独](https://leetcode-cn.com/problems/sudoku-solver/)等。
在《[回溯算法详解](https://labuladong.gitbook.io/algo/di-ling-zhang-bi-du-xi-lie/hui-su-suan-fa-xiang-jie-xiu-ding-ban)》一文中提到,解决一个回溯问题,实际上就是一个决策树的遍历过程,需要思考三个问题:
(1)路径:已经做出的选择。
(2)选择列表:当前可以做的选择。
(3)结束条件:到达决策树底层,无法再做选择的条件。
下面是改编过的算法通用结构。
~~~
function backtrack(路径, 选择列表):
if 满足结束条件
console.log(路径)
return
for 选择 of 选择列表
做选择
backtrack(路径, 选择列表)
撤销选择
~~~
面试题12[矩阵路径](https://leetcode-cn.com/problems/ju-zhen-zhong-de-lu-jing-lcof/)和面试题13[机器人运动范围](https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof/)。在二维方格或矩阵的运动可用回溯法解决。
## 一、N皇后
[N皇后](https://leetcode-cn.com/problems/n-queens/)是一道经典的回溯算法题,将 n 个皇后放置在 n×n 的棋盘上,使皇后彼此之间不能相互攻击,即每个棋子所在的行、列、对角线都不能有另一个棋子。
在下面的[示例](https://codepen.io/strick/pen/JjGzMed)中,N是皇后的数量,backtrack()函数是回溯过程(如下所列),isValid()函数判断是否符合选中条件。
(1)从第一个 row=0 开始。
(2)循环列并且试图在每个 column 中放置皇后。
(3)如果方格 (row, column) 在攻击范围内,那么跳过。
(4)在 (row, column) 方格上放置皇后,继续寻找下一个位置。
(5)判断 row 是否和皇后数量相同。
~~~
const N = 4;
function backtrack(route, row) {
if (row == N) { //结束条件
console.log(route);
return;
}
for (let column = 0; column < N; column++) {
if (!isValid(route, row, column))
continue;
route[row] = column; //做选择
backtrack(route, row + 1); //下一步
route[row] = null; //撤销选择(可省略)
}
}
//从下往上 判断row行column列放置是否合适
function isValid(route, row, column) {
let leftup = column - 1,
rightup = column + 1;
for (let i = row - 1; i >= 0; i--) { // 逐行往上考察每一行
if (route[i] == column) // 第i行的column列有棋子
return false;
if (leftup >= 0) {
if (route[i] == leftup) // 考察左上对角线:第i行leftup列有棋子
return false;
}
if (rightup < N) {
if (route[i] == rightup) // 考察右上对角线:第i行rightup列有棋子
return false;
}
leftup--;
rightup++;
}
return true;
}
~~~
## 二、0-1背包
有一个背包,背包总的承载重量是 Wkg。现在有 n 个物品,假设每个物品的重量都不相等,并且不可分割。期望选择几件物品,装载到背包中。在不超过背包容量的前提下,如何让背包中物品的总重量最大?
把物品依次排列,对于物品选择装或不装,然后递归余下的物品,[如下所示](https://codepen.io/strick/pen/dyGrmzw)。
~~~
let max = Number.MIN_VALUE,
W = 100;
function backtrack(route, goods) {
let weight = route.length ? route.reduce((acc, cur) => acc += cur) : 0;
if (weight == W || route.length == goods.length) { //结束条件
if (weight > max && weight <= W) {
max = weight;
}
console.log(route);
return;
}
for (let i = 0; i < goods.length; i++) {
if(weight + goods[i] > W || route.indexOf(goods[i]) > -1)
continue;
route.push(goods[i]); //做选择
backtrack(route, goods);
route.pop(); //撤销选择
}
}
~~~
## 三、全排列
[全排列](https://leetcode-cn.com/problems/permutations/)是指输出给定数字序列的全部可能的排列,假设序列中的数字都是唯一的,利用回溯算法枚举出所有排列,[如下所示](https://codepen.io/strick/pen/VweNBKd)。
~~~
function backtrack(route, nums) {
if (route.length == nums.length) { //结束条件
console.log(route);
return;
}
for (let i = 0; i < nums.length; i++) {
if (route.indexOf(nums[i]) > -1)
continue;
route.push(nums[i]); //做选择
backtrack(route, nums);
route.pop(); //撤销选择
}
}
~~~
面试题17[打印从 1~n 位的数](https://leetcode-cn.com/problems/da-yin-cong-1dao-zui-da-de-nwei-shu-lcof/)。将问题转换成数字排列,用递归实现。
*****
> 原文出处:
[博客园-数据结构和算法躬行记](https://www.cnblogs.com/strick/category/1809992.html)
已建立一个微信前端交流群,如要进群,请先加微信号freedom20180706或扫描下面的二维码,请求中需注明“看云加群”,在通过请求后就会把你拉进来。还搜集整理了一套[面试资料](https://github.com/pwstrick/daily),欢迎阅读。
![](https://docs.gechiui.com/gc-content/uploads/sites/kancloud/2e1f8ecf9512ecdd2fcaae8250e7d48a_430x430.jpg =200x200)
';