《剑指offer》刷题系列——（四十一）矩阵中的路径

题目

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的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

思路

两个for循环的含义：
从头开始对矩阵中的每个字符进行深度优先搜索，一旦找到一条符合条件的路径，则返回True，不再搜索。

递归参数： 当前元素在矩阵 board 中的行列索引 i 和 j ，当前目标字符在 word 中的索引 k 。
终止条件：
返回 false ： ① 行或列索引越界 或 ② 当前矩阵元素与目标字符不同 或 ③ 当前矩阵元素已访问过 （③ 可合并至 ② ） 。
返回 true： 字符串 word 已全部匹配，即 k = len(word) - 1 。

递推工作：
（1）标记当前矩阵元素： 将 board[i][j] 值暂存于变量 tmp ，并修改为字符 ‘/’ ，代表此元素已访问过，防止之后搜索时重复访问
（2）搜索下一单元格： 朝当前元素的 上、下、左、右 四个方向开启下层递归，使用 或 连接 （代表只需一条可行路径） ，并记录结果至 res 。
（3） 还原当前矩阵元素： 将 tmp 暂存值还原至 board[i][j] 元素。

回溯返回值： 返回 res ，代表是否搜索到目标字符串。

代码

方法一：

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        for i in range(len(board)):
            for j in range(len(board[0])):
                if self.dfs(board,word,i,j,0):
                    return True
        return False
        
    def dfs(self,board,word,i,j,k):
        if not 0<=i<len(board) or not 0<=j<len(board[0]) or board[i][j]!=word[k]: return False
        if k==len(word)-1:
            return True
        temp = board[i][j]
        board[i][j] = ' '
        res = self.dfs(board,word,i-1,j,k+1) or self.dfs(board,word,i+1,j,k+1) or self.dfs(board,word,i,j-1,k+1) or self.dfs(board,word,i,j+1,k+1)
        board[i][j]=temp
        return res

方法二：借助二维数组标记走过的路径

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        flag = [[0 for _ in range(len(board[0]))] for _ in range(len(board))]
        for i in range(len(board)):
            for j in range(len(board[0])):
                if self.rec(i,j,board,word,0,flag):
                    return True
        return False

    def rec(self,i,j,board,word,k,flag):
        if i<0 or i>=len(board) or j<0 or j>=len(board[0]) or word[k]!=board[i][j]: return False
        if flag[i][j]==1: return False
        if k==len(word)-1: return True
        
        flag[i][j]=1
        res = self.rec(i+1,j,board,word,k+1,flag) or self.rec(i-1,j,board,word,k+1,flag) or self.rec(i,j+1,board,word,k+1,flag) or self.rec(i,j-1,board,word,k+1,flag)
        flag[i][j]=0
        return res

复杂度

参考：面试题12. 矩阵中的路径（深度优先搜索 DFS ，清晰图解）