请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的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
输入:board = [[“a”,“b”],[“c”,“d”]], word = “abcd”
输出:false
思路:遍历board矩阵中的每一个节点,找到能和word[0]对应上的那个节点,然后从该节点DFS即可
代码:
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
row, col = len(board), len(board[0])
word_lenth = len(word)
#k是word中的第几个元素
def dfs(i, j, k):
if i >= row or i < 0 or j >= col or j < 0 or board[i][j] != word[k]:return False
#如果k == word_lenth - 1,说明已经在board中匹配到了所有的word的元素,返回True
if k == word_lenth - 1:return True
#将board[i][j]置为空格,防止其被重复访问
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(row):
for j in range(col):
if dfs(i, j, 0):
return True
return False