AI智能
改变未来

3、回溯算法解题套路框架——Go语言版

前情提示:Go语言学习者。本文参考https://labuladong.gitee.io/algo,代码自己参考抒写,若有不妥之处,感谢指正

关于golang算法文章,为了便于下载和整理,都已开源放在:

  • https://github.com/honlu/GoLabuladongAlgorithm
  • https://gitee.com/dreamzll/GoLabuladongAlgorithm方便就请分享,star!备注转载地址!欢迎一起学习和交流!

涉及题目

  • 46. 全排列(中等)
  • 51. N皇后(困难)

回溯算法其实就是我们常说的 DFS 算法,本质上就是一种暴力穷举算法。

废话不多说,直接上回溯算法框架。解决一个回溯问题,实际上就是一个决策树的遍历过程。你只需要思考 3 个问题:

1、路径:也就是已经做出的选择。

2、选择列表:也就是你当前可以做的选择。

3、结束条件:也就是到达决策树底层,无法再做选择的条件。

如果你不理解这三个词语的解释,没关系,我们后面会用「全排列」和「N 皇后问题」这两个经典的回溯算法问题来帮你理解这些词语是什么意思,现在你先留着印象。

代码方面,回溯算法的框架:

// 伪码var res [][]intfunc backTrack(路径,选择列表){if 满足结束条件{ // 终止条件res = append(res, 路径)  // 存放结果return}for _,选择 := range 选择列表{ // 选择:本层集合中元素(树中节点孩子的数量就是集合的大小)做选择  // 处理节点backTrack(路径,选择列表)  // 递归撤销选择  // 回溯,撤销处理结果}}

其核心就是 for 循环里面的递归,在递归调用之前「做选择」,在递归调用之后「撤销选择」,特别简单。

什么叫做选择和撤销选择呢,这个框架的底层原理是什么呢?下面我们就通过「全排列」这个问题来解开之前的疑惑,详细探究一下其中的奥妙!

一、全排列问题

我们在高中的时候就做过排列组合的数学题,我们也知道

n

个不重复的数,全排列共有

n!

个。

PS:为了简单清晰起见,我们这次讨论的全排列问题不包含重复的数字

那么我们当时是怎么穷举全排列的呢?比方说给三个数

[1,2,3]

,你肯定不会无规律地乱穷举,一般是这样:

先固定第一位为 1,然后第二位可以是 2,那么第三位只能是 3;然后可以把第二位变成 3,第三位就只能是 2 了;然后就只能变化第一位,变成 2,然后再穷举后两位……

其实这就是回溯算法,我们高中无师自通就会用,或者有的同学直接画出如下这棵回溯树:

只要从根遍历这棵树,记录路径上的数字,其实就是所有的全排列。我们不妨把这棵树称为回溯算法的「决策树」

为啥说这是决策树呢,因为你在每个节点上其实都在做决策。比如说你站在下图的红色节点上:

你现在就在做决策,可以选择 1 那条树枝,也可以选择 3 那条树枝。为啥只能在 1 和 3 之中选择呢?因为 2 这个树枝在你身后,这个选择你之前做过了,而全排列是不允许重复使用数字的。

现在可以解答开头的几个名词:

[2]

就是「路径」,记录你已经做过的选择;

[1,3]

就是「选择列表」,表示你当前可以做出的选择;「结束条件」就是遍历到树的底层,在这里就是选择列表为空的时候

如果明白了这几个名词,可以把「路径」和「选择」列表作为决策树上每个节点的属性,比如下图列出了几个节点的属性:

我们定义的

backtrack

函数其实就像一个指针,在这棵树上游走,同时要正确维护每个节点的属性,每当走到树的底层,其「路径」就是一个全排列

再进一步,如何遍历一棵树?这个应该不难吧。回忆一下之前 学习数据结构的框架思维 写过,各种搜索问题其实都是树的遍历问题,而多叉树的遍历框架就是这样:

// 伪码func traverse(root *TreeNode){for _, child := range root.children{// 前序遍历需要的操作traverse(child)// 后序遍历需要的操作}}

而所谓的前序遍历和后序遍历,他们只是两个很有用的时间点,我给你画张图你就明白了:

前序遍历的代码在进入某一个节点之前的那个时间点执行,后序遍历代码在离开某个节点之后的那个时间点执行

回想我们刚才说的,「路径」和「选择」是每个节点的属性,函数在树上游走要正确维护节点的属性,那么就要在这两个特殊时间点搞点动作:

现在,你是否理解了回溯算法的这段核心框架?

var res [][]int// 主函数,输入一组不重复的数字,返回他们的全排列func permute(nums []int) [][]int {// 记录"路径"res = [][]int{}var track []intbackTrack(nums, track)return res}// 路径:记录在track中// 选择列表:nums中不存在于track的那些元素// 结束条件:nums中的元素全都在track中出现func backTrack(nums []int, track []int) {if len(nums) == 0 {p := make([]int, len(track))copy(p, track) // 学习新函数res = append(res, p)}for i := 0; i < len(nums); i++ {// 做选择cur := nums[i]track = append(track, nums[i])nums = append(nums[:i],nums[i+1:]...)//直接使用切片,删除nums[i]// 进入下一层决策树backTrack(nums, track)// 取消选择,删除最后一个元素nums = append(nums[:i],append([]int{cur},nums[i:]...)...)//回溯的时候切片也要复原,元素位置不能变track = track[:len(track)-1]}}

我们这里稍微做了些变通,没有显式记录「选择列表」,而是通过

nums

track

推导出当前的选择列表:

至此,我们就通过全排列问题详解了回溯算法的底层原理。当然,这个算法解决全排列不是很高效,应为对链表使用

contains

方法需要 O(N) 的时间复杂度。有更好的方法通过交换元素达到目的,但是难理解一些,这里就不写了,有兴趣可以自行搜索一下。

但是必须说明的是,不管怎么优化,都符合回溯框架,而且时间复杂度都不可能低于 O(N!),因为穷举整棵决策树是无法避免的。这也是回溯算法的一个特点,不像动态规划存在重叠子问题可以优化,回溯算法就是纯暴力穷举,复杂度一般都很高

明白了全排列问题,就可以直接套回溯算法框架了,下面简单看看 N 皇后问题。

二、N 皇后问题

这个问题很经典了,简单解释一下:给你一个 N×N 的棋盘,让你放置 N 个皇后,使得它们不能互相攻击。

PS:皇后可以攻击同一行、同一列、左上左下右上右下四个方向的任意单位。

这个问题本质上跟全排列问题差不多,决策树的每一层表示棋盘上的每一行;每个节点可以做出的选择是,在该行的任意一列放置一个皇后。

直接套用回溯算法框架:

import "strings"var res [][]string// 输入棋盘边长n, 返回所有合法的放置方法func solveNQueens(n int) [][]string {res = [][]string{}// 棋盘定义和初始化board := make([][]string, n)for i := 0; i < n; i++ {board[i] = make([]string, n)}for i := 0; i < n; i++ {for j := 0; j < n; j++ {board[i][j] = "."}}// 回溯backTrack(board, 0)return res}// 路径:board中小于row的那些行都已经成功放置了皇后// 选择列表:第row行的所有列都是放置皇后的选择// 结束条件:row超过board的最后一行,说明棋盘满了func backTrack(board [][]string, row int) {// 触发结束条件size := len(board)if row == size {temp := make([]string, size) // 保存一种棋盘解法for i := 0; i < size; i++ {  // 将棋盘n*n 转为n*1的数组temp[i] = strings.Join(board[i], "") // 将第i行的结果,连接起来}res = append(res, temp)return}for col := 0; col < size; col++ {// 排除不合法选择if !isValid(board, row, col) {continue}// 做选择board[row][col] = "Q"// 进入下一行决策backTrack(board, row+1)// 撤销选择board[row][col] = "."}}

这部分主要代码,其实跟全排列问题差不多。

isValid

函数的实现也很简单:

// 判断是否可以在board[row][col]放置皇后func isValid(board [][]string, row, col int) bool {n := len(board)// 检查列中是否有皇后互相冲突for i := 0; i < row; i++ {if board[i][col] == "Q" {return false}}// 检查右上方是否有皇后互相冲突for i, j := row-1, col+1; i >= 0 && j < n; i, j = i-1, j+1 {if board[i][j] == "Q" {return false}}// 检查左上方是否有皇后互相冲突for i, j := row-1, col-1; i >= 0 && j >= 0; i, j = i-1, j-1 {if board[i][j] == "Q" {return false}}return true}

PS:肯定有读者问,按照 N 皇后问题的描述,我们为什么不检查左下角,右下角和下方的格子,只检查了左上角,右上角和上方的格子呢?

因为皇后是一行一行从上往下放的,所以左下方,右下方和正下方不用检查(还没放皇后);因为一行只会放一个皇后,所以每行不用检查。也就是最后只用检查上面,左上,右上三个方向。

函数

backtrack

依然像个在决策树上游走的指针,通过

row

col

就可以表示函数遍历到的位置,通过

isValid

函数可以将不符合条件的情况剪枝:

如果直接给你这么一大段解法代码,可能是懵逼的。但是现在明白了回溯算法的框架套路,还有啥难理解的呢?无非是改改做选择的方式,排除不合法选择的方式而已,只要框架存于心,你面对的只剩下小问题了。

N = 8

时,就是八皇后问题,数学大佬高斯穷尽一生都没有数清楚八皇后问题到底有几种可能的放置方法,但是我们的算法只需要一秒就可以算出来所有可能的结果。

不过真的不怪高斯。这个问题的复杂度确实非常高,看看我们的决策树,虽然有

isValid

函数剪枝,但是最坏时间复杂度仍然是 O(N^(N+1)),而且无法优化。如果

N = 10

的时候,计算就已经很耗时了。

有的时候,我们并不想得到所有合法的答案,只想要一个答案,怎么办呢?比如解数独的算法,找所有解法复杂度太高,只要找到一种解法就可以。

其实特别简单,只要稍微修改一下回溯算法的代码即可:

func backTrack(board [][]string, row int) bool{// 触发结束条件size := len(board)if row == size {temp := make([]string, size) // 保存一种棋盘解法for i := 0; i < size; i++ {  // 将棋盘n*n 转为n*1的数组temp[i] = strings.Join(board[i], "") // 将第i行的结果,连接起来}res = append(res, temp)return true}for col := 0; col < size; col++ {// 排除不合法选择if !isValid(board, row, col) {continue}// 做选择board[row][col] = "Q"// 进入下一行决策if backTrack(board, row+1){return true}// 撤销选择board[row][col] = "."}}

这样修改后,只要找到一个答案,for 循环的后续递归穷举都会被阻断。也许你可以在 N 皇后问题的代码框架上,稍加修改,写一个解数独的算法?

三、最后总结

回溯算法就是个多叉树的遍历问题,关键就是在前序遍历和后序遍历的位置做一些操作,算法框架如下:

func backTrack(参数) {if (终止条件) {存放结果;return;}for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {处理节点;backTrack(路径,选择列表); // 递归回溯,撤销处理结果}}

backtrack

函数时,需要维护走过的「路径」和当前可以做的「选择列表」,当触发「结束条件」时,将「路径」记入结果集

其实想想看,回溯算法和动态规划是不是有点像呢?我们在动态规划系列文章中多次强调,动态规划的三个需要明确的点就是「状态」「选择」和「base case」,是不是就对应着走过的「路径」,当前的「选择列表」和「结束条件」?

某种程度上说,动态规划的暴力求解阶段就是回溯算法。只是有的问题具有重叠子问题性质,可以用 dp table 或者备忘录优化,将递归树大幅剪枝,这就变成了动态规划。而今天的两个问题,都没有重叠子问题,也就是回溯算法问题了,复杂度非常高是不可避免的。

赞(0) 打赏
未经允许不得转载:爱站程序员基地 » 3、回溯算法解题套路框架——Go语言版