劍指offer 面試12題

面試12題：

題目：矩陣中的路徑

題：請設計一個函數，用來判斷在一個矩陣中是否存在一條包含某字符串所有字符的路徑。路徑可以從矩陣中的任意一個格子開始，每一步可以在矩陣中向左，向右，向上，向下移動一個格子。如果一條路徑經過了矩陣中的某一個格子，則該路徑不能再進入該格子。 例如 a b c e s f c s a d e e 矩陣中包含一條字符串"bcced"的路徑，但是矩陣中不包含"abcb"路徑，因為字符串的第一個字符b占據了矩陣中的第一行第二個格子之后，路徑不能再次進入該格子。

 

解題思路：回溯法，使用遞歸。

 

解題代碼：

# -*- coding:utf-8 -*-
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