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

优质解答: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

浙公网安备 33010602011771号