题目:
给定一个包含非负整数的 m x n 网格 grid ,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。
说明:每次只能向下或者向右移动一步。
思路:
动态规划,dp[i][j]:从grid[0][0]到grid[i][j]的最小总和
- 先对dp数组的第0行和第0列,用前缀和进行初始化
- 按行遍历,dp[i][j]=min(dp[i][j-1],dp[i-1][j])+grid[i][j]
注:每次只能向下或者向右移动一步。故当前格子一定是由其上或其左的格子转移而来的,故只要知道其上和其左的格子的最小路径总和,即可知道当前格子的最小路径总和。
解答:
class Solution:
def minPathSum(self, grid: List[List[int]]) -> int:
#动态规划
m=len(grid)
n=len(grid[0])
#dp[i][j]:从grid[0][0]到grid[i][j]的最小总和
dp=[[0]*n for _ in range(m)]
for i in range(m):
if i==0:
dp[i][0]=grid[0][0]
continue
dp[i][0]=dp[i-1][0]+grid[i][0]
for j in range(n):
if j==0:
dp[0][j]=grid[0][0]
continue
dp[0][j]=dp[0][j-1]+grid[0][j]
for i in range(1,m):
for j in range(1,n):
dp[i][j]=min(dp[i][j-1],dp[i-1][j])+grid[i][j]
return dp[m-1][n-1]