剑指Offer(Python多种思路实现):矩阵中的路径

剑指Offer(Python多种思路实现):矩阵中的路径

面试12题：

题目：矩阵中的路径

题：请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bcced"的路径，但是矩阵中不包含"abcb"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

解题思路一：回溯法，使用递归。

class Solution:
    def hasPath(self, matrix, rows, cols, path):
        # write code here
        if not matrix or rows<0 or cols<0 or path==None:
            return False
        markmatrix=[0]*(rows*cols)
        pathIndex=0

        for row in range(rows):
            for col in range(cols):
                if self.hasPathCore(matrix,rows,cols,row,col,path,pathIndex,markmatrix):
                    return True
        return False

    def hasPathCore(self,matrix,rows,cols,row,col,path,pathIndex,markmatrix):
        if pathIndex==len(path):
            return True
        hasPath=False
        if row>=0 and row<rows and col>=0 and col<cols and matrix[row*cols+col]==path[pathIndex] and not markmatrix[row*cols+col]:
            pathIndex+=1
            markmatrix[row*cols+col]=True
            hasPath=self.hasPathCore(matrix,rows,cols,row+1,col,path,pathIndex,markmatrix) or \
                    self.hasPathCore(matrix, rows, cols, row-1, col,path, pathIndex, markmatrix) or \
                    self.hasPathCore(matrix, rows, cols, row, col+1, path, pathIndex, markmatrix) or \
                    self.hasPathCore(matrix, rows, cols, row, col-1, path, pathIndex, markmatrix)
            if not hasPath:
                pathIndex -=1
                markmatrix[row*cols+col]=False
        return hasPath

解题思路二：

class Solution:
    def hasPath(matrix: str, rows: int, cols: int, path: str) -> 'bool':
        for i in range(rows):   # 双重循环找矩阵元素
            for j in range(cols):
                if matrix[i*cols + j] == path[0]:
                    if spread(list(matrix), rows, cols, path[1:], i, j):
                        return True
        return False
    
    def spread(matrix: str, rows: int, cols: int, path: str, i: int, j: int) -> 'bool': # 这里的matrix是复制矩阵
        if not path:
            return True
        matrix[i*cols + j] = '-'    # 已访问，覆盖
        spreaded = False
        for x, y in ((i - 1, j), (i + 1, j), (i, j - 1), (i, j + 1)):
            if 0<=x<rows and 0<=y<cols and matrix[x*cols + y] == path[0]:
                if spread(matrix, rows, cols, path[1:], x, y):
                    spreaded = True
        return spreaded