剑指offer65.矩阵中的路径

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sinat_36811967/article/details/87931894

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

回溯，参考leetcode79. 单词搜索，注意这里给的matrix和path都是字符串：

# -*- coding:utf-8 -*-
class Solution:
    def hasPath(self, matrix, rows, cols, path):
        # write code here
        def helper(count, locs, a, b):
            if count == len(path):
                return True
            if a<0 or a>=rows or b<0 or b>=cols:
                return False
            if [a, b] not in locs and path[count]==matrix[a*cols+b]:
                return helper(count+1, locs+[[a,b]], a+1, b) or \
                       helper(count+1, locs+[[a,b]], a-1, b) or \
                       helper(count+1, locs+[[a,b]], a, b+1) or \
                       helper(count+1, locs+[[a,b]], a, b-1)
            return False
        for i in range(rows):
            for j in range(cols):
                if helper(0, [], i, j):
                    return True
        return False