题目:https://leetcode-cn.com/problems/unique-paths/
一个机器人位于一个 m x n 网格的左上角 (起始点在下图中标记为 “Start” )。
机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角(在下图中标记为 “Finish” )。
问总共有多少条不同的路径?
示例 1:
输入:m = 3, n = 7
输出:28
示例 2:
输入:m = 3, n = 2
输出:3
解释:
从左上角开始,总共有 3 条路径可以到达右下角。
1. 向右 -> 向右 -> 向下
2. 向右 -> 向下 -> 向右
3. 向下 -> 向右 -> 向右
示例 3:
输入:m = 7, n = 3
输出:28
示例 4:
输入:m = 3, n = 3
输出:6
提示:
1 <= m, n <= 100
题目数据保证答案小于等于 2 * 109
注:示例2解释可看出m和n写反了,m=2,n=3
解题思路:
经典动态规划,转移方程dp[i][j] = dp[i-1][j]+dp[i][j-1],其中dp[i][j]表示走到(i,j)位置上的路径有多少条。转移方程一般从最后一步倒推,即写出走最后一步的方程。由于机器人每次只能向下或向右移动一步,所以要走到(i,j),上一步一定是(i-1,j)或(i,j-1),走到(i,j)的路径数dp[i][j]就等于走到(i-1,j)的路径数dp[i-1][j]与走到(i,j-1)的路径数dp[i][j-1]之和。注意dp是二维数组,所以一般初始值不是个别一两个值,而是边界一行值或一列值。机器人在走dp[i][0]和dp[0][j]时都只有一种走法(要么全向下,要么全向右),所以初始值dp[i][0]=1、dp[0][j]=1。
代码:
class Solution:
def uniquePaths(self, m: int, n: int) -> int:
dp = [[0] * n for _ in range(m)]
for i in range(m):
for j in range(n):
if(i == 0):
dp[i][j] = 1
elif(j == 0):
dp[i][j] = 1
else:
dp[i][j] = dp[i - 1][j] + dp[i][j - 1]
return dp[m - 1][n - 1]
测试:
if __name__ == "__main__":
m = 3
n = 3
res = Solution().uniquePaths(m, n)
print(res)