LeetCode 每日一题(2020.7.23)

题目地址(64.最小路径和)

https://leetcode-cn.com/problems/minimum-path-sum/

题目描述

给定一个包含非负整数的 m x n 网格,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。
说明:每次只能向下或者向右移动一步。

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

Python代码

class Solution:
    def minPathSum(self, grid: List[List[int]]) -> int:
    	if not grid or not grid[0]:
    		return 0
    	r = len(grid)
    	c = len(grid[0])
    	dp = [[0] * c for _ in range(r)]
    	dp[0][0] = grid[0][0]
    	for i in range(1,r):
    		dp[i][0] = dp[i - 1][0] + grid[i][0]
    	for j in range(1,c):
    		dp[0][j] = dp[0][j - 1] + grid[0][j]
    	for i in range(1,r):
    		for j in range(1,c):
    			dp[i][j] = min(dp[i - 1][j], dp[i][j - 1]) + grid[i][j]
    	return dp[-1][-1]

空间优化后的Python代码

class Solution:
    def minPathSum(self, grid):
        dp = [float('inf')] * (len(grid[0])+1)
        dp[1] = 0
        for row in grid:
            for idx, num in enumerate(row):
                dp[idx + 1] = min(dp[idx], dp[idx + 1]) + num
        return dp[-1]

猜你喜欢

转载自blog.csdn.net/hitxqy/article/details/107548950