剑指offer面试题【12】----矩阵中的路径【回溯法】【Python】

题目描述

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

代码实现

# -*- coding:utf-8 -*-
class Solution:
    mask=[]
    def hasPath(self, matrix, rows, cols, path):
        # write code here
        m_s="".join(matrix) #将数组化为字符串，容易查找
        count=m_s.count(path[0]) #有几个可能的起点
        t=0
        k=0
        if count==0:
            return False
        else:
            while k<count:
                #判断是否重复
                mask=[[0]*cols for i in range(rows)] #创建二维数组，并初始化为0
                index=m_s.find(path[0],t)#find(str, beg=0, end=len(string)) 方法检测字符串中是否包含子字符串 str ，如果指定 beg（开始） 和 end（结束） 范围，则检查是否包含在指定范围内，如果包含子字符串返回开始的索引值，否则返回-1。
                i=index/cols
                j=index%cols
                t=index+1 #控制下一个起点可能的搜索范围
                k=k+1
                if self.define(matrix,rows,cols,i,j,path,mask):
                    return True
            return False
    def define(self,matrix,rows,cols,i,j,path,mask):
        if len(path)==0:
            return True
        #连通域判定条件
        if i>=0 and i<=rows-1 and j>=0 and j<=cols-1 and matrix[i*cols +j]==path[0] and mask[i][j]!=1:
            mask[i][j]=1#表示已经过该节点
            copy=path[1:]#将字符串右移，下次递归将判断下一个字符串
            #递归，看下一个位置和下一个字符是否满足条件，只要有一个满足，则返回真
            return self.define(matrix,rows,cols,i-1,j,copy,mask) or \
                self.define(matrix,rows,cols,i+1,j,copy,mask) or \
                self.define(matrix,rows,cols,i,j-1,copy,mask) or \
                self.define(matrix,rows,cols,i,j+1,copy,mask)
        #不满足连通域条件，返回False
        return False