8.19 - hard - 67

329. Longest Increasing Path in a Matrix

九章算法班里講過的一道題,用記憶化搜索。

class Solution(object):
    def longestIncreasingPath(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: int
        """
        if not matrix:
            return 0
        # write your code here
        A = matrix
        max_length = 0
        dp = [[0 for _ in range(len(A[0]))] for _ in range(len(A))]
        visited = [[False for _ in range(len(A[0]))] for _ in range(len(A))]
        for i in range(len(A)):
            for j in range(len(A[0])):
                max_length = max(max_length, self.dfs(dp, visited, A, i, j))
        
        return max_length
                
    def dfs(self, dp, visited, A, i, j):
        if visited[i][j]:
            return dp[i][j] # starting from i,j the longest subsequence
        
        length = 1
        for x, y in [[i-1, j], [i+1, j], [i, j-1], [i, j+1]]:
            if 0 <= x < len(A) and 0 <= y < len(A[0]) and A[x][y] < A[i][j]:
                length = max(length, self.dfs(dp, visited, A, x, y) + 1)
        
        dp[i][j] = length
        visited[i][j] = True
        return length
            
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容