请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。例如,在下面的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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ju-zhen-zhong-de-lu-jing-lcof
提示:
1 <= board.length <= 200
1 <= board[i].length <= 200
DFS递归,一般找路径用DFS,找最短之类的用BFS
先写row,col是因为DFS里面要用到,保证上下左右移动后的ni,nj不越界
递归终止条件是吧word字符串中所有的字符都走完,cur是从第二个字符,index=1进入递归,则当cur指向字符串最后一个字符时,再+1进入下一层递归,但此时cur指针不指向任何字符串,因为已经指完了,此时cur==len(word),直接return,回到指向最后字符的那层递归。
class Solution:
def exist(self, board: List[List[str]], word: str) -> bool:
row = len(board)
col = len(board[0])
def DFS(i,j,cur,visited):
if cur == len(word):
return True
pos = ((0,-1),(-1,0),(1,0),(0,1))
for m,n in pos:
ni,nj = i+m,j+n
# 满足不越界,且坐标没访问过且坐标值等于字符串中的下一字符,继续递归
if 0<=ni<row and 0<=nj<col and (ni,nj) not in visited and board[ni][nj]==word[cur]:
# 这里如果不加这个判断返回true,则会返回for循环外的False
if DFS(ni,nj,cur+1,visited+[(ni,nj)]):
return True
#如果返回False,返回上一层递归看有没有其他路径满足,如果都没有就出递归返回False
return False
# DFS如果返回False就再找其他入口,其他入口也不行就返回False
for i in range(row):
for j in range(col):
if board[i][j]==word[0]: #找到入口
if DFS(i,j,1,[(i,j)]):
return True
return False
牛客上这个矩阵的路径题的输入给的不是矩阵是个字符串,得自己加一句:
matrix = [list(matrix[cols*i:cols*(i+1)]) for i in range(rows)]
牛客版本:
# -*- coding:utf-8 -*-
class Solution:
def hasPath(self, matrix, rows, cols, path):
# write code here
def DFS(i,j,cur,visited):
if cur == len(path):
return True
pos = ((0,-1),(-1,0),(1,0),(0,1))
for m,n in pos:
ni,nj = i+m,j+n
# 满足不越界,且坐标没访问过且坐标值等于字符串中的下一字符,继续递归
if 0<=ni<rows and 0<=nj<cols and (ni,nj) not in visited and matrix[ni][nj]==path[cur]:
# 这里如果不加这个判断返回true,则会返回for循环外的False
if DFS(ni,nj,cur+1,visited+[(ni,nj)]):
return True
#如果返回False,返回上一层递归看有没有其他路径满足,如果都没有就出递归返回False
return False
# DFS如果返回False就再找其他入口,其他入口也不行就返回False
matrix = [list(matrix[cols*i:cols*i+cols]) for i in range(rows)]
for i in range(rows):
for j in range(cols):
if matrix[i][j]==path[0]: #找到入口
if DFS(i,j,1,[(i,j)]):
return True
return False