Leetcode 62. 不同路径【动态规划】

Leetcode 62. 不同路径

中等题，动态规划的经典题目

题目

一个机器人位于一个 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

解法

数学方法

这就是一个排列组合问题，总共走(m+n-2)步，其中向下走(m-1)步，向右走(n-1)步，那么就在总步数里面挑向下走的步数，也就是$C_{m+n-1}^{m-1}$。
按照数学计算代入
$$\begin{array}{rl}{C}_M^m& =A[m,M]/m!\\ A_M^m& =M\ast (M-1)\ast ...\ast (M-m+1)\\ A_M^m& =M!/(M-m)!\\ C_M^m& =M!/(M-m)!/m!\end{array}$$

class Solution:
    def uniquePaths(self, m: int, n: int) -> int:
        x, y = 1, 1 
        for i in range(m+n-2,m-1,-1): x *= i
        for i in range(1,n): y *= i
        return x//y

或者直接用阶乘math.factorial()求

class Solution:
    def uniquePaths(self, m: int, n: int) -> int:
        return math.factorial(m+n-2)//math.factorial(m-1)//math.factorial(n-1)

• 时间复杂度 O(min(m,n))
• 空间复杂度 O(1)

动态规划

定义dp数组表示到当前位置的路径数，则有递推公式dp[i][j] = dp[i-1][j]+dp[i][j-1]，因为只能从上方或者左方到达ij位置。

class Solution:
    def uniquePaths(self, m: int, n: int) -> int:
        dp = [[0 for _ in range(n+1)] for _ in range(m+1)]
        dp[1][1] = 1
        for i in range(1,m+1):
            for j in range(1,n+1):
                if i==1 and j==1: continue
                dp[i][j] = dp[i-1][j]+dp[i][j-1]
        return dp[m][n] 

• 时间复杂度 O(mn)
• 空间复杂度 O(mn)
  由于dp[i][j]只与上方和左侧有关，因此遍历到第i行时，可以直接用dp[j]表示第i行的j位置路径数，未更新时dp[j]即为上方信息。
  递推公式为dp[j] = dp[j]+dp[j-1],，由此可以将空间压缩到一维数组。

class Solution:
    def uniquePaths(self, m: int, n: int) -> int:
        #空间压缩
        dp = [0 for _ in range(n+1)]
        dp[1] = 1 #初始的原点为1
        for i in range(1,m+1):
            for j in range(1,n+1): 
                dp[j] = dp[j]+dp[j-1]
        return dp[n]

• 时间复杂度 O(mn)
• 空间复杂度 O(n)