题目
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[[“a”,“b”,“c”,“e”],
[“s”,“f”,“c”,“s”],
[“a”,“d”,“e”,“e”]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
示例 1:
输入:board = [[“A”,“B”,“C”,“E”],[“S”,“F”,“C”,“S”],[“A”,“D”,“E”,“E”]], word = “ABCCED”
输出:true
示例 2:
输入:board = [[“a”,“b”],[“c”,“d”]], word = “abcd”
输出:false
思路
深度优先搜索,对每一个元素执行依次。分别从上下左右进行深度优先搜索,如果索引超出范围或者矩阵元素与字符串当前元素不相等,则返回False,如果相等且当前字符串索引为最后一个,则返回True
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
def DFS(row, col, k):
if not 0<= row < len(board) or not 0 <= col < len(board[0]) or board[row][col] != word[k]:
return False
if k == len(word) - 1:
return True
board[row][col], tmp = '/', board[row][col]
ans = DFS(row+1,col,k+1) or DFS(row-1,col,k+1) or DFS(row,col-1,k+1) or DFS(row,col+1,k+1)
board[row][col] = tmp
return ans
for row in range(len(board)):
for col in range(len(board[0])):
if DFS(row, col, 0): return True
return False