题目描述:
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。 例如 a b c e s f c s a d e e 这样的3 X 4 矩阵中包含一条字符串”bcced”的路径,但是矩阵中不包含”abcb”路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
思路:本题是使用回溯法的典型题目。由于无法确定路径在字符矩阵中的起点,因此在矩阵中任选一格作为起点,进行搜索,判断是否存在所求路径。另外还需要与字符矩阵同样大小的标志数组,用来检查当前位置字符是否已经包含在路径中。矩阵中找到的第i个字符也一定对应着路径中第i个位置的字符
# -*- coding:utf-8 -*-
class Solution:
def hasPath(self, matrix, rows, cols, path):
# write code here
if len(matrix)==0 or len(matrix)!=rows*cols or len(path)==0:
return False
visited=[False]*len(matrix);
haspath1=0;
for i in range(rows):
for j in range(cols):
if self.haspath(matrix,rows,cols,path,i,j,visited,haspath1):
return True
return False
def haspath(self,mattrix,rows,cols,path,i,j,visited,haspath1):
# matrix: 字符矩阵
# rows: 矩阵的行数
# cols: 矩阵的列数
# path: 需要寻找的路径
# i: 当前位置的横坐标(对应行数)
# j: 当前位置的纵坐标(对应列数)
# visited: 访问标志数组
# pathlength: 已经找到的路径长度
# :return:是否存在路径
if haspath1==len(path):
return True
curhaspath=False
if (0<=i<rows and 0<=j<cols) and mattrix[i*cols+j]==path[haspath1] and visited[i*cols+j]==False:
visited[i*cols+j]=True;
haspath1+=1
curhaspath=self.haspath(mattrix,rows,cols,path,i-1,j,visited,haspath1) or \
self.haspath(mattrix,rows,cols,path,i+1,j,visited,haspath1) or \
self.haspath(mattrix, rows, cols, path, i , j-1, visited, haspath1) or \
self.haspath(mattrix, rows, cols, path, i , j+1, visited, haspath1)
if not curhaspath:
haspath1-=1
visited[i*cols+j] = False
return curhaspath
m=Solution()
print(m.hasPath(["a","b","t","g","c","f","c","s"],2,4,"bf"))
m=["a","b","t","g","c","f","c","s"]
print(m[1*4+1])