矩阵中的路径

Posted 2021-03-04 isshpan

题目要求：

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始，每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格，那么该路径不能再次进入该格子。例如，在下面的3×4的矩阵中包含一条字符串“bfce”的路径（路径中的字母用加粗标出）。
[["a","b","c","e"],
["s","f","c","s"],
["a","d","e","e"]]
但矩阵中不包含字符串“abfb”的路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入这个格子。

算法分析

dfs遍历路径，遍历过的将其值写为None表示已遍历，记得最后恢复状态（dfs不熟可以作为模板，多写就熟了）。

代码

class Solution(object):
    def exist(self, board, word):
        for row in range(len(board)):
            for col in range(len(board[0])):
                if self.dfs(board, word, row, col):
                    return True
        return False
    def dfs(self, board, word, row, col):
        if len(word) == 0:
            return True
        if row >= len(board) or row < 0 or col >= len(board[0]) or col < 0 or board[row][col] is None or board[row][col] != word[0]:
            return False
        x = board[row][col]
        board[row][col] = None
        flag = self.dfs(board, word[1:], row + 1, col) or self.dfs(board, word[1:], row - 1, col) or self.dfs(board, word[1:], row, col + 1) or self.dfs(board, word[1:], row, col - 1)
        board[row][col] = x # 恢复状态
        return flag