请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的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
解题思路
采用深度有限搜索。从一个元素开始,按照上下左右的顺序进行深度优先遍历,成功则返回True;不成功则回溯至上一个节点。
剪枝。当当前路径无法匹配成功时(如元素不同或该元素已经被访问),则立刻返回,回溯至上一个节点。
代码
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
self.board = board
self.word = word
for i in range(len(board)): # 按行
for j in range(len(board[0])):
if self.dfs(i,j,0):
return True
return False # 遍历完后仍不能返回True,则返回False
def dfs(self, i,j,k): # 第i行第j列,是否匹配第k个字符
board = self.board
word = self.word
if not (0 <= i < len(board)) or not (0 <= j < len(board[0])) or (board[i][j] != word[k]): # 不匹配的情况
return False
# 匹配的情况
if k == len(word) - 1: # 如果k是最后一个字符,则返回True
return True
tmp, board[i][j] = board[i][j], '/' # 如果不是最后一个字符,有必要将这个字符暂时置为'/',以防止重复
res = self.dfs(i+1,j,k+1) or self.dfs(i-1,j, k+1) or self.dfs(i, j+1, k+1) or self.dfs(i, j-1, k+1) # 对上下左右依次进行搜索
board[i][j] = tmp # 搜索完成后,将board[i][j]的值复原
return res
总结
1. Python可以使用 <=i<=这样的判断语句;
2. 二维list a和a[0]的含义.
3. 类中的函数f调用自身递归需要加上self.f