LeetCode-Python-64. 最小路径和

给定一个包含非负整数的 m x n 网格,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。

说明:每次只能向下或者向右移动一步。

示例:

输入:
[
  [1,3,1],
  [1,5,1],
  [4,2,1]
]
输出: 7
解释: 因为路径 1→3→1→1→1 的总和最小。

思路:

动态规划。

dp[m][n] = min(dp[m - 1][n], dp[m][n - 1]) + grid[m][n]。注意考虑边界条件。

class Solution(object):
    def minPathSum(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        #dp[m][n] = min(dp[m - 1][n], dp[m][n - 1]) + grid[m][n]
        if not grid:
            return 0
        m = len(grid)
        n = len(grid[0])
        dp = [[0] * (n)] * (m) 
        for i in range(m):
            for j in range(n):
                # print dp, i, j
                if not i and not j:
                    dp[i][j] = grid[i][j]
                elif i and not j:
                    dp[i][j] = dp[i-1][j] + grid[i][j]
                elif not i and j:
                    dp[i][j] = dp[i][j-1] + grid[i][j]
                else:
                    dp[i][j] = min(dp[i-1][j], dp[i][j-1]) + grid[i][j]
        return dp[m-1][n-1]

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/88064398