剑指 Offer 12. 矩阵中的路径

题目:剑指 Offer 12. 矩阵中的路径

image

优质解答:DFS+剪枝(参考自K神)

从矩阵中任一点作为起点进行深度优先遍历,矩阵内可向四个方向移动,遍历到不符合要求的元素(矩阵越界以及字符不匹配)则进行剪枝,并将走过的点先置空,防止走重复的路径,在回溯过程中再将其恢复,如果判定到达字符串最后一位且字符匹配则直接返回True。

class Solution:
    def exist(self, board: List[List[str]], word: str) -> bool:
        def dfs(i, j, k):
            if not (0 <= i < len(board)) or not (0 <= j <len(board[0])) or board[i][j] != word[k]: # 越界及字符串不匹配直接返回false剪枝,不在从此往下遍历
                return False
            if k == len(word) - 1:   return True
            board[i][j] = ''		# 将该位置空,防止从下个节点往回遍历
            res = dfs(i+1, j, k+1) or dfs(i-1, j, k+1) or dfs(i, j+1, k+1) or dfs(i, j-1, k+1) # 遍历四个方向,寻找可行解
            board[i][j] = word[k] # 进行回溯并将该位恢复
            return res

        for i in range(len(board)):
            for j in range(len(board[0])):
                if dfs(i, j, 0):
                    return True
        return False
posted @ 2022-04-04 16:17  面包房主人  阅读(21)  评论(0)    收藏  举报