给定一个包含非负整数的 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] <= 200
思路:做的题多了,一看到这道题就反应过来大概率是要用动态规划来解决更好。
首先定义dp数组,dp[i][j]表示达到位置(i,j)的最短路径的长度。
然后思考递推公式,题目规定每一步只能向下或者向右移动,因此dp[i][j]的值只跟dp[i-1][j]和dp[i][j-1]有关,然后我们要的是最短路径,肯定是取二者中最小的那个,不过别忘了,走到这个位置还要加上grid[i][j]的值。那么递推公式就出来了:dp[i][j] = min(dp[i-1][j],dp[i][j-1])+grid[i][j]。
然后考虑初始化:dp[i][j]的值只跟dp[i-1][j]和dp[i][j-1]有关,因此我们想要知道dp[1][1]的值,必须先定义dp[1][0]和dp[0][1],可以观察发现,第一行的任何一个位置只能由它左边的那个位置向右移动一步得到;同理第一列的任何一个位置只能由它上面的那个位置向下移动一步得到,因此我们可以初始化第一行和第一列的值。
最后是遍历顺序,很直观,一定是从左往右,从上往下。
代码:(Python)
class Solution(object):
def minPathSum(self, grid):
m = len(grid)
n = len(grid[0])
dp = [[0]* n for _ in range(m)]
dp[0][0] = grid[0][0]
for i in range(1,m): #初始化第一列
dp[i][0] = grid[i][0] + dp[i-1][0]
for j in range(1,n): #初始化第一行
dp[0][j] = grid[0][j] + dp[0][j-1]
for i in range(1,m):
for j in range(1,n):
dp[i][j] = min(dp[i-1][j],dp[i][j-1])+grid[i][j]
return dp[-1][-1]