leetcode931. 下降路径最小和

最新推荐文章于 2024-11-04 13:51:18 发布

996冲冲冲

最新推荐文章于 2024-11-04 13:51:18 发布

阅读量50

点赞数

分类专栏：动态规划文章标签：动态规划算法

本文链接：https://blog.csdn.net/weixin_44392348/article/details/128835654

版权

动态规划专栏收录该内容

39 篇文章 0 订阅

订阅专栏

基础动态规划
base case:
dp[0][i] = matrix[0][i]
状态转移记住有三个方向还有注意不要越界

def minFallingPathSum(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: int
        """
        m,n = len(matrix),len(matrix[0])
        dp = [[22222]*n for i in range(m)]
        for i in range(n):
            dp[0][i] = matrix[0][i]
        for i in range(1,m):
            for j in range(0,n):
                if j-1 >= 0 and j+1 < n:
                    dp[i][j] = min(dp[i-1][j],dp[i-1][j-1],dp[i-1][j+1]) + matrix[i][j]
                elif j-1 < 0:
                     dp[i][j] = min(dp[i-1][j],dp[i-1][j+1]) + matrix[i][j]
                else:
                    dp[i][j] = min(dp[i-1][j],dp[i-1][j-1]) + matrix[i][j]
        return min(dp[-1])