最小路径和

题目描述

来源于 https://leetcode-cn.com/

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

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

示例:

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

解法:

class Solution:
    def minPathSum(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        m = len(grid)
        n = len(grid[0])
        
        inf = float('inf')
        
        for col in range(n-1, -1, -1):
            for row in range(m-1, -1, -1):
                if row < m-1:
                    down = grid[row+1][col]
                else:
                    down = inf
                    
                if col < n-1:
                    right = grid[row][col+1]
                else:
                    right = inf
                
                if down != inf or right != inf:
                    grid[row][col] += min(right, down)
            
        return grid[0][0]