[剑指offer]矩阵中的路径
描述:题目来源:牛客网剑指offer JZ65 题
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如
[
a
b
c
e
s
f
c
s
a
d
e
e
]
\begin{bmatrix} a & b & c &e \\ s & f & c & s \\ a & d & e& e\\ \end{bmatrix}
⎣⎡asabfdcceese⎦⎤
矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
示例1
输入:
[[a,b,c,e],[s,f,c,s],[a,d,e,e]],"abcced"
返回值:
true
示例2
输入:
[[a,b,c,e],[s,f,c,s],[a,d,e,e]],"abcb"
返回值:
false
备注:
0 <= matrix.length <= 200
0 <= matrix[i].length <= 200
题解:
首先在矩阵中找到与首字母对应的位置,然后通过深度优先搜索算法,分别向该点的上下左右进行搜索匹配,在匹配过程中将前面匹配的点记录下来,防止多次记录的情况。下面是Python代码:
class Solution:
def hasPath(self , matrix , word ):
# write code here
visited = [[0 for i in range(len(matrix[0]))] for j in range(len(matrix))]
for i in range(len(matrix)):
for j in range(len(matrix[0])):
if self.DFS(i, j, matrix, visited, word):
return True
return False
def DFS(self, x, y, matrix, visited, word):
if not word:
return True
if not visited[x][y]:
if matrix[x][y] == word[0]:
if len(word[1:])==0:
return True
visited[x][y] = 1
if y+1 <= len(matrix[0])-1 and self.DFS(x,y+1, matrix, visited, word[1:]):
return True
if x+1 <= len(matrix)-1 and self.DFS(x+1, y, matrix, visited, word[1:]):
return True
if x-1 >= 0 and self.DFS(x-1, y, matrix, visited, word[1:]):
return True
if y-1 >= 0 and self.DFS(x, y-1, matrix, visited, word[1:]):
return True
visited[x][y] = 0
return False