矩阵中的路径

题目描述

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

思路：

深度优先搜索加递归实现

深度优先：

首先每个格子只能移动一格，也就是行列坐标只有一个发生改变，加1或者-1；

使用矩阵大小的记忆单元都设置成True,表示都为使用的格子；

 

递归步骤：

若所给字符串为空，则返回True;

从上到下，从左到右，开始遍历矩阵，寻找path中的第一个字符，找不到返回False；

找得到设置相应位置的记忆矩阵中的单元为False,表示此格被搜索过，然后继续递归调用dfs寻找字符串之后的字符；

 

代码：

# -*- coding:utf-8 -*-
class Solution:
    def hasPath(self, matrix, rows, cols, path):
        # write code here
        def dfs(memories,r,c,s):
            if s=='':
                return True
            dx=[-1,1,0,0]
            dy=[0,0,-1,1]
            for k in range(4):
                x=dx[k]+r
                y=dy[k]+c
                if x>=0 and x<rows and y>=0 and y<cols and memories[x][y] and matrix[x*cols+y]==s[0]:
                    memories[x][y]=False
                    #递归调用返回True时说明有路径可达
                    if dfs(memories[:],x,y,s[1:]):
                        return True
                    #无路径可达时，还将记忆单元中的位置设置为True,以便下次还可通过
                    memories[x][y]=True
            return False
                    
            
        if path=='':
            return True
        memories=[[True for c in range(cols)] for r in range(rows)]
        for r in range(rows):
            for c in range(cols):
                if matrix[r*cols+c]==path[0]:
                    memories[r][c]=False
                    if dfs(memories[:],r,c,path[1:]):
                        return True
                    memories[r][c]=True
        return False