Python 动态规划解决不同路径问题

最新推荐文章于 2024-10-19 10:10:17 发布

原创

最新推荐文章于 2024-10-19 10:10:17 发布 · 2.5k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #动态规划

一、【LeetCode 62】不同路径

1. 题目描述

一个机器人位于一个 m x n 网格的左上角（起始点在下图中标记为“Start” ）。

机器人每次只能向下或者向右移动一步。机器人试图达到网格的右下角（在下图中标记为“Finish”）。

问总共有多少条不同的路径？
在这里插入图片描述
例如，上图是一个7 x 3 的网格。有多少可能的路径？

示例 1:

输入: m = 3, n = 2
输出: 3
解释:
从左上角开始，总共有 3 条路径可以到达右下角。
1. 向右 -> 向右 -> 向下
2. 向右 -> 向下 -> 向右
3. 向下 -> 向右 -> 向右

示例 2:

输入: m = 7, n = 3
输出: 28

提示：
1 <= m, n <= 100
题目数据保证答案小于等于 2 * 10 ^ 9

2. 解题思路

这是一个经典的动态规划问题，关键是要想明白如何去计算其中的某一个格子到达它的路径有多少条，计算方式就是到达它上面的格子的路径数加上到达它左边的格子的路径数。

首先初始化一个二维数组memo（m行n列，每个元素的初始值为0）memo = [[0 for j in range(n)] for i in range(m)]
接着对memo数组的第一行和第一列进行初始化：因为机器人每次只能向下或者向右移动一步，不能向左向上往回走，所以可以将 memo 数组的第一行和第一列的值全部初始化为1（因为到达第一行的每个格子只有一直往右走这一种走法，同理到达第一列的每个格子也只有一直往下走这一种走法）
接下来就是双重循环，考虑数组内部的值，每一个格子的值都等于上一个格子的值加上左边格子的值memo[i][j] = memo[i - 1][j] + memo[i][j - 1]，可见下图：
这样依次算出memo数组的每一个值后，直接返回右下角的数值即可return memo[-1][-1]。

3. 代码

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