剑指Offer - 矩阵中的路径 -- Python实现

题目描述

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

 

思路：回溯法

找到一个满足第一个字符的，递归找剩余字符串的第一个字符是否能满足。
若不能满足，提前return False 回溯剪枝。

# -*- coding:utf-8 -*-
class Solution:
    def hasPath(self, matrix, rows, cols, path):
        # write code here
        for i in range(rows):
            for j in range(cols):
                if matrix[i*cols + j] == path[0]:
                    if self.find_path(list(matrix), rows, cols, path[1:], i, j):
                    # 表示上面返回True时，匹配成功。但是返回False时并不一定匹配失败
                    # 只是说明当前第一个和path[0]相同的位置匹配失败，后面可能还有匹配成功的位置！！！
                    # 故不能添加 else：return False
                        return True

    def find_path(self, matrix, rows, cols, path, i, j):
        if not path:  # 若剩余的path为空，则匹配成功
            return True
        matrix[i*cols + j] = ''  # 此位置字符置空，防止第二次被用到
        # 然后接着找剩余字符串的第一个字符是否能和上一个匹配成功的位置的上下左右的字符匹配成功。
        if j+1 < cols and matrix[i*cols+j+1] == path[0]:
            return self.find_path(matrix, rows, cols, path[1:], i, j+1)
        elif j-1 >= 0 and matrix[i*cols+j-1] == path[0]:
            return self.find_path(matrix, rows, cols, path[1:], i, j-1)
        elif i+1 < rows and matrix[(i+1)*cols+j] == path[0]:
            return self.find_path(matrix, rows, cols, path[1:], i+1, j)
        elif i-1 >= 0 and matrix[(i-1)*cols+j] == path[0]:
            return self.find_path(matrix, rows, cols, path[1:], i-1, j)
        else:
            return False  # 回溯剪枝