python-leetcode329-深度优先搜索矩阵中的最长递增路径

Posted 2020-11-27 西西嘛呦

给定一个整数矩阵，找出最长递增路径的长度。

对于每个单元格，你可以往上，下，左，右四个方向移动。 你不能在对角线方向上移动或移动到边界外（即不允许环绕）。

示例 1:

输入: nums =
[
[9,9,4],
[6,6,8],
[2,1,1]
]
输出: 4
解释: 最长递增路径为 [1, 2, 6, 9]。
示例 2:

输入: nums =
[
[3,4,5],
[3,2,6],
[2,2,1]
]
输出: 4
解释: 最长递增路径是 [3, 4, 5, 6]。注意不允许在对角线方向上移动。

class Solution:
    def longestIncreasingPath(self, matrix):
        if not matrix:
            return 0
        row_n = len(matrix)
        col_n = len(matrix[0])
        res = 1
        resut_save = [[0 for _ in range(col_n)] for _ in range(row_n)]
        for i in range(row_n):
            for j in range(col_n):
                res = max(res, self.dfs(matrix, i, j, resut_save))
        return res
    def dfs(self,matrix, i, j, resut_save):
        if resut_save[i][j] != 0:
            return  resut_save[i][j]
        dirctions = [(1, 0), (-1, 0), (0, 1), (0, -1)]
        res = 1
        for dx, dy in dirctions:
            x = i + dx
            y = j + dy
            if 0<= x < len(matrix) and 0<= y < len(matrix[0]) and matrix[x][y] > matrix[i][j]:
                res = max(res, self.dfs(matrix, x, y, resut_save) + 1)
        resut_save[i][j] = res
        return res

s=Solution()
res=s.longestIncreasingPath([[9,9,4],[6,6,8],[2,1,1]])
print(res)