聊聊DP入门之不同路径

开发 前端
本文分别给出了深搜,动规,数论三种方法。深搜当然是超时了,顺便分析了一下使用深搜的时间复杂度,就可以看出为什么超时了。然后在给出动规的方法,依然是使用动规五部曲,这次我们就要考虑如何正确的初始化了,初始化和遍历顺序其实也很重要!

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

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

问总共有多少条不同的路径?

示例 1:

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

示例 2:

  • 输入:m = 2, n = 3
  • 输出:3

解释:从左上角开始,总共有 3 条路径可以到达右下角。

  • 向右 -> 向右 -> 向下
  • 向右 -> 向下 -> 向右
  • 向下 -> 向右 -> 向右

示例 3:

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

示例 4:

  • 输入:m = 3, n = 3
  • 输出:6

提示:

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

思路

深搜

这道题目,刚一看最直观的想法就是用图论里的深搜,来枚举出来有多少种路径。

注意题目中说机器人每次只能向下或者向右移动一步,那么其实机器人走过的路径可以抽象为一颗二叉树,而叶子节点就是终点!

如图举例:

不同路径

此时问题就可以转化为求二叉树叶子节点的个数,代码如下:

  1. class Solution { 
  2. private: 
  3.     int dfs(int i, int j, int m, int n) { 
  4.         if (i > m || j > n) return 0; // 越界了 
  5.         if (i == m && j == n) return 1; // 找到一种方法,相当于找到了叶子节点 
  6.         return dfs(i + 1, j, m, n) + dfs(i, j + 1, m, n); 
  7.     } 
  8. public
  9.     int uniquePaths(int m, int n) { 
  10.         return dfs(1, 1, m, n); 
  11.     } 
  12. }; 

大家如果提交了代码就会发现超时了!

来分析一下时间复杂度,这个深搜的算法,其实就是要遍历整个二叉树。

这颗树的深度其实就是m+n-1(深度按从1开始计算)。

那二叉树的节点个数就是 2^(m + n - 1) - 1。可以理解深搜的算法就是遍历了整个满二叉树(其实没有遍历整个满二叉树,只是近似而已)

所以上面深搜代码的时间复杂度为,可以看出,这是指数级别的时间复杂度,是非常大的。

动态规划

机器人从(0 , 0) 位置出发,到(m - 1, n - 1)终点。

按照动规五部曲来分析:

确定dp数组(dp table)以及下标的含义

dp[i][j] :表示从(0 ,0)出发,到(i, j) 有dp[i][j]条不同的路径。

确定递推公式

想要求dp[i][j],只能有两个方向来推导出来,即dp[i - 1][j] 和 dp[i][j - 1]。

此时在回顾一下 dp[i - 1][j] 表示啥,是从(0, 0)的位置到(i - 1, j)有几条路径,dp[i][j - 1]同理。

那么很自然,dp[i][j] = dp[i - 1][j] + dp[i][j - 1],因为dp[i][j]只有这两个方向过来。

dp数组的初始化

如何初始化呢,首先dp[i][0]一定都是1,因为从(0, 0)的位置到(i, 0)的路径只有一条,那么dp[0][j]也同理。

所以初始化代码为:

  1. for (int i = 0; i < m; i++) dp[i][0] = 1; 
  2. for (int j = 0; j < n; j++) dp[0][j] = 1; 

确定遍历顺序

这里要看一下递归公式dp[i][j] = dp[i - 1][j] + dp[i][j - 1],dp[i][j]都是从其上方和左方推导而来,那么从左到右一层一层遍历就可以了。

这样就可以保证推导dp[i][j]的时候,dp[i - 1][j] 和 dp[i][j - 1]一定是有数值的。

举例推导dp数组

如图所示:

不同路径

以上动规五部曲分析完毕,C++代码如下:

  1. class Solution { 
  2. public
  3.     int uniquePaths(int m, int n) { 
  4.         vector<vector<int>> dp(m, vector<int>(n, 0)); 
  5.         for (int i = 0; i < m; i++) dp[i][0] = 1; 
  6.         for (int j = 0; j < n; j++) dp[0][j] = 1; 
  7.         for (int i = 1; i < m; i++) { 
  8.             for (int j = 1; j < n; j++) { 
  9.                 dp[i][j] = dp[i - 1][j] + dp[i][j - 1]; 
  10.             } 
  11.         } 
  12.         return dp[m - 1][n - 1]; 
  13.     } 
  14. }; 

时间复杂度:

空间复杂度:

其实用一个一维数组(也可以理解是滚动数组)就可以了,但是不利于理解,可以优化点空间,建议先理解了二维,在理解一维,C++代码如下:

  1. class Solution { 
  2. public
  3.     int uniquePaths(int m, int n) { 
  4.         vector<int> dp(n); 
  5.         for (int i = 0; i < n; i++) dp[i] = 1; 
  6.         for (int j = 1; j < m; j++) { 
  7.             for (int i = 1; i < n; i++) { 
  8.                 dp[i] += dp[i - 1]; 
  9.             } 
  10.         } 
  11.         return dp[n - 1]; 
  12.     } 
  13. }; 

时间复杂度:

空间复杂度:

数论方法

在这个图中,可以看出一共m,n的话,无论怎么走,走到终点都需要 m + n - 2 步。

不同路径

在这m + n - 2 步中,一定有 m - 1 步是要向下走的,不用管什么时候向下走。

那么有几种走法呢?可以转化为,给你m + n - 2个不同的数,随便取m - 1个数,有几种取法。

那么这就是一个组合问题了。

那么答案,如图所示:

不同路径

求组合的时候,要防止两个int相乘溢出! 所以不能把算式的分子都算出来,分母都算出来再做除法。

例如如下代码是不行的。

  1. class Solution { 
  2. public
  3.     int uniquePaths(int m, int n) { 
  4.         int numerator = 1, denominator = 1; 
  5.         int count = m - 1; 
  6.         int t = m + n - 2; 
  7.         while (count--) numerator *= (t--); // 计算分子,此时分子就会溢出 
  8.         for (int i = 1; i <= m - 1; i++) denominator *= i; // 计算分母 
  9.         return numerator / denominator; 
  10.     } 
  11. }; 

需要在计算分子的时候,不断除以分母,代码如下:

  1. class Solution { 
  2. public
  3.     int uniquePaths(int m, int n) { 
  4.         long long numerator = 1; // 分子 
  5.         int denominator = m - 1; // 分母 
  6.         int count = m - 1; 
  7.         int t = m + n - 2; 
  8.         while (count--) { 
  9.             numerator *= (t--); 
  10.             while (denominator != 0 && numerator % denominator == 0) { 
  11.                 numerator /= denominator; 
  12.                 denominator--; 
  13.             } 
  14.         } 
  15.         return numerator; 
  16.     } 
  17. }; 

时间复杂度:

空间复杂度:

计算组合问题的代码还是有难度的,特别是处理溢出的情况!

总结

本文分别给出了深搜,动规,数论三种方法。

深搜当然是超时了,顺便分析了一下使用深搜的时间复杂度,就可以看出为什么超时了。

然后在给出动规的方法,依然是使用动规五部曲,这次我们就要考虑如何正确的初始化了,初始化和遍历顺序其实也很重要!

 

责任编辑:武晓燕 来源: 代码随想录
相关推荐

2022-01-10 11:28:55

数据结构算法DP入门

2021-09-30 11:55:00

微服务

2022-01-11 10:01:25

二叉搜索树数量

2010-06-10 15:36:23

路由协议的分类

2021-12-28 07:20:44

斐波那契数算法数字

2021-05-07 08:02:53

Sentinel 流量服务

2009-12-31 10:03:58

VPN配置实例

2021-09-30 09:58:14

路径总和二叉树

2023-06-05 12:59:03

2021-12-29 11:32:38

数据结构算法爬楼梯

2021-09-01 22:58:22

Canvas标签

2009-12-21 15:04:02

路由器配置

2021-07-11 12:12:49

.NETJWTjson

2022-12-28 08:16:16

metric聚合java

2022-03-02 07:52:13

React类组件函数式组件

2020-05-27 08:05:33

MybatisMapper接口

2021-06-08 09:28:12

.Net通知服务

2021-08-26 17:36:42

Floyd算法数据结构

2016-11-25 00:45:37

队列数据

2016-11-28 08:40:17

系统降级服务
点赞
收藏

51CTO技术栈公众号