题目:判断一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵汇总的任意一格开始,每一步可以在矩阵中向左、右、上、下移动一格。如果一条路径经过了矩阵的某一格,那么该路径不能再次进入该格子。
def has_path(matrix,string):
rows=len(matrix)
cols=len(matrix[0])
# 已经访问过的矩阵元素
visited=[[0 for i in range(cols)] for j in range(rows)]
# 已找到字符的个数
pathlength = 0
row=0
# 遍历矩阵,寻找string中的第一个字符,如果有一条匹配即存在,跳出
while row<rows:
col=0
while col<cols:
if path(matrix,row,col,string,pathlength,visited):
# print(row,col)
return True
col+=1
row+=1
return False
def path(matrix,row,col,string,pathlength,visited):
rows = len(matrix)
cols = len(matrix[0])
# print(row,col)
if pathlength==len(string):
return True
haspath = False
# print(row,col,string[pathlength],visited[row][col])
if row>=0 and col>=0 and row<rows and col<cols and matrix[row][col] == string[pathlength] and visited[row][col]==0:
# print(row,col,pathlength)
pathlength+=1
visited[row][col]=1
# print(visited)
haspath = path(matrix,row,col-1,string,pathlength,visited) or path(matrix,row-1,col,string,pathlength,visited) or \
path(matrix,row,col+1,string,pathlength,visited) or path(matrix,row+1,col,string,pathlength,visited)
if not haspath:
pathlength-=1
visited[row][col]=0
return haspath
注:
遍历矩阵中的每个字符,每次调用回溯函数,进行判断。如果存在,返回True。
回溯函数中止条件为,查找到字符个数等于目标字符串长度。如果查找到当前字符为目标字符串中相应的字符,则向上下左右递归查找。如果该字符上下左右找不到下一个字符,回溯上一步。