剑指 Offer 12. 矩阵中的路径 [中等]

剑指 Offer 12. 矩阵中的路径 [中等]

题目描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的3×4的矩阵中包含一条字符串“bfce”的路径（路径中的字母用加粗标出）。

[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]

但矩阵中不包含字符串“abfb”的路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入这个格子。

示例:

输入：board = [["A","B","C","E"],["S","F","C","S"],["A","D","E","E"]], word = "ABCCED"
输出：true

输入：board = [["a","b"],["c","d"]], word = "abcd"
输出：false

限制：

1 <= board.length <= 200
1 <= board[i].length <= 200

解题思路：

本问题是典型的矩阵搜索问题，可使用 深度优先搜索（DFS）+ 剪枝 解决。

• 深度优先搜索： 可以理解为暴力法遍历矩阵中所有字符串可能性。DFS 通过递归，先朝一个方向搜到底，再回溯至上个节点，沿另一个方向搜索，以此类推。
• 剪枝： 在搜索中，遇到 这条路不可能和目标字符串匹配成功 的情况（例如：此矩阵元素和目标字符不同、此元素已被访问），则应立即返回，称之为 可行性剪枝 。

DFS 解析：

• 递归参数： 当前元素在矩阵 board 中的行列索引 i 和 j ，当前目标字符在 word 中的索引 k 。
• 终止条件：
  • 返回 false ： (1) 行或列索引越界 或 (2) 当前矩阵元素与目标字符不同 或 (3) 当前矩阵元素已访问过 （ (3) 可合并至 (2) ） 。
  • 返回 true ： k = len(word) - 1 ，即字符串 word 已全部匹配。
• 递推工作：
  • 标记当前矩阵元素： 将 board[i][j] 修改为 空字符 '' ，代表此元素已访问过，防止之后搜索时重复访问。
  • 搜索下一单元格： 朝当前元素的 上、下、左、右 四个方向开启下层递归，使用 或 连接 （代表只需找到一条可行路径就直接返回，不再做后续 DFS ），并记录结果至 res 。
  • 还原当前矩阵元素： 将 board[i][j] 元素还原至初始值，即 word[k] 。
• 返回值： 返回布尔量 res ，代表是否搜索到目标字符串。

（1）

（2）

（3）

（4）

（5）

（6）

（7）

（8）

复杂度分析：

代码：

package main

import "fmt"

func exist(board [][]byte, word string) bool {
	rows := len(board)
	cols := len(board[0])
	for i := 0; i < rows; i++ {
		for j := 0; j < cols; j++ {
			if dfs(i, j, board, word, 0) {
				return true
			}
		}
	}
	return false
}

func dfs(i, j int, board [][]byte, word string, k int) bool {
	fmt.Println(i,j,board)
	if i < 0 || i >= len(board) || j < 0 || j >= len(board[0]) || board[i][j] != word[k] {
		return false
	}
	if k == (len(word) - 1) {
		return true
	}
	board[i][j] = ' ' // 标记成空元素防止重复访问
	// 上方
	top := dfs(i-1, j, board, word, k+1)
	// 下方
	down := dfs(i+1, j, board, word, k+1)
	// 左方
	left := dfs(i, j-1, board, word, k+1)
	// 右方
	right := dfs(i, j+1, board, word, k+1)

	res := top || down || left || right // 为了方便理解用了中间变量，这里可以直接调dfs，减少小内存分配
	board[i][j] = word[k] // 元素还原至初始值，即 word[k]，因为走到自己这里board[i][j]初始值必定等于word[k]
	return res
}

func main() {
	test := [][]byte{{'A', 'B', 'C', 'E'}, {'S', 'F', 'C', 'S'}, {'A', 'D', 'E', 'E'}}
	fmt.Println(exist(test, "ABCCED"))
}