给定一个包含非负整数的 m x n 网格 grid ,请找出一条从左上角到右下角的路径,使得路径上的数字总和为最小。
说明:每次只能向下或者向右移动一步。
示例 1:
输入:grid = [[1,3,1],[1,5,1],[4,2,1]]
输出:7
解释:因为路径 1→3→1→1→1 的总和最小。
示例 2:
输入:grid = [[1,2,3],[4,5,6]]
输出:12
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 200
0 <= grid[i][j] <= 100
题解:
class Solution:
def minPathSum(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid[0])
# 只有一个单元格时
if m == 1 and n == 1:
return grid[0][0]
dp = [[0] * n for i in range(m)]
for i in range(m):
for j in range(n):
if i == 0:
# 第一行,和 = 当前单元格数字 + 前一个单元格数字
dp[i][j] = dp[i][j - 1] + grid[i][j]
elif j == 0:
# 第一列,和 = 当前单元格数字 + 上一个单元格数字
dp[i][j] = dp[i - 1][j] + grid[i][j]
else:
# 中间单元格,和 = 当前单元格数字 + 上一个和前一个单元格中最小的数字
dp[i][j] = min(dp[i][j - 1], dp[i - 1][j]) + grid[i][j]
return dp[m-1][n-1]
注意:
dp[i][j - 1]、dp[i - 1][j]
代表的是状态转移,它包括前面走过的路径之和,而grid[i][j]
代表的仅仅只是当前单元格一个数字,所以是:dp[i][j] = dp[i][j - 1] + grid[i][j]
,而不是dp[i][j] = grid[i][j - 1] + grid[i][j]