难度中等230收藏分享切换为英文接收动态反馈
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的3×4的矩阵中包含一条字符串“bfce”的路径(路径中的字母用加粗标出)。
[["a","b","c","e"],
["s","f","c","s"],
["a","d","e","e"]]
但矩阵中不包含字符串“abfb”的路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入这个格子。
DSF
dfs可以用栈实现,即将该点的所有临近点加入栈,再弹出栈顶,并对栈顶重复此操作
可以递归实现,visit该点,并遍历所有临近点,每次dfs(该点)
这里采用第二种方式
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
left = len(board[0])-1
low = len(board)-1
def dfs(i, j, k):
if not (0<=i<=low) or not(0<=j<=left) or board[i][j] != word[k]:
return False
if k == len(word)-1:
return True
board[i][j] = 0
z = 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 z
for i in range(len(board)):
for j in range(len(board[0])):
# print(i, j)
if dfs(i, j, 0):
return True
return False
在dfs函数中的判断条件:
当当前点超界或者当前点不符合(断掉),弹回上一层递归 fasle
当当前点为最后一个,返回true
又因为只要找到一条就行,所以return 的 z 为几条路径的or ,即有一条就行
注意点 1
一开始写的 bfs 中有board参数,即
def borad(i, j, board, k)
发现在该函数下对board改变时,函数外board也会改变
于是采用深拷贝
import copy
def borad(i, j, board, k):
board = copy.deepcopy(board)
注意点 2
深拷贝耗时很大,所以想办法只用board这一个二维列表
board[i][j] = 0
z = 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]
如上代码,经过了i j 这个点,将其置为0,传给其下的点,这样其下的dfs中的board就为置0的
但是不能改变大局,即弹出时该点不能为0,所以在将他置回去即可
注意点 3
一开始写的时候,在当前点判断下一个点是否超界,即
z1, z2, z3, z4 = False, False, False, False
if i-1 >= 0 and board[i-1][j] != 0:
z1 = dfs(i-1, j, k+1)
if j-1 >= 0 and board[i][j-1] != 0:
z2 = dfs(i, j-1, k+1)
if i+1 <= low and board[i+1][j] != 0:
z3 = dfs(i+1, j, k+1)
if j+1 <= left and board[i][j+1] != 0:
z4 = dfs(i, j+1, k+1)
但是这样很繁琐,我们尽量将判断放到bfs一开始,即当前点判断当前点的条件