剑指offer1.跳台阶 & 力扣70.爬楼梯 && leetcode 剑指 Offer 10- II. 青蛙跳台阶问题

70. 爬楼梯

假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢？
注意：给定 n 是一个正整数。示例 1：

输入： 2
输出： 2
解释： 有两种方法可以爬到楼顶。
1.  1 阶 + 1 阶
2.  2 阶

示例 2：

输入： 3
输出： 3
解释： 有三种方法可以爬到楼顶。
1.  1 阶 + 1 阶 + 1 阶
2.  1 阶 + 2 阶
3.  2 阶 + 1 阶

思路：动态规划

经典的动态规划题
dp[i] = dp[i - 1] + dp[i - 2],

初值dp[1] = 1, dp[2] = 2;

 1 class Solution {
 2     public int climbStairs(int n) {
 3         if(n == 1 || n == 2){
 4             return n;
 5         }
 6         int[] dp = new int[n + 1];
 7         dp[1] = 1;
 8         dp[2] = 2;
 9         for(int i = 3; i <= n; i++){
10             dp[i] = dp[i-1] + dp[i-2];
11         }
12         return dp[n];
13     }
14 }

力扣测试时间为0ms，空间为36.4MB

复杂度分析：

时间复杂度：O(n)

空间复杂度：O(n)

扩展改进

由于一直累加的过程中可能会超过int32或者int64的范围，所以要利用求余运算规则 (x%k + y%k)%k = (x+y)%k

另外由于dp[i]只与前两项有关，所以可以使用两个临时变量一直迭代即可，省去了O(N)的空间复杂度

比如下面这题：

剑指 Offer 10- II. 青蛙跳台阶问题

一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

答案需要取模 1e9+7(1000000007)，如计算初始结果为：1000000008，请返回 1。

示例 1：

输入：n = 2
输出：2

示例 2：

输入：n = 7
输出：21

示例 3：

输入：n = 0
输出：1

提示：

0 <= n <= 100

思路：

思路参考：https://leetcode-cn.com/problems/qing-wa-tiao-tai-jie-wen-ti-lcof/solution/mian-shi-ti-10-ii-qing-wa-tiao-tai-jie-wen-ti-dong/

如果继续使用上面的解法，最终的结果会因为大整数越界而出现错误，可以尝试使用求余运算规则 (x%k + y%k)%k = (x+y)%k

 1 class Solution {
 2     public int numWays(int n) {
 3         // 其实就是一个斐波那契数列
 4         // 动态规划，由于dp[i]只与前两项有关，所以可以使用两个临时变量一直迭代即可，省去了O(N)的空间复杂度
 5         // 由于一直累加的过程中可能会超过int32或者int64的范围，所以要利用求余运算规则 (x%k + y%k)%k = (x+y)%k
 6         // dp[i] = dp[i-1] + dp[i-2] ==> dp[i] % k = (dp[i-1]%k + dp[i-2]%k)%k
 7 
 8         int a = 1, b = 1, sum = 1;
 9         for(int i = 0; i < n; i++){
10             sum = (a + b) % 1000000007;
11             a = b;
12             b = sum;
13         }
14         return a;
15     }
16 }

leetcode运行时间为0ms -100%, 空间为35.7 MB - 33.85%

复杂度分析：

时间复杂度：迭代n次，所以时间复杂度为O(n)

空间复杂度：只有几个临时变量，所以空间复杂度为O(1)