题目
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
示例 1:
输入: 2
输出: 2
解释: 有两种方法可以爬到楼顶。
- 1 阶 + 1 阶
- 2 阶
示例 2:
输入: 3
输出: 3
解释: 有三种方法可以爬到楼顶。
- 1 阶 + 1 阶 + 1 阶
- 1 阶 + 2 阶
- 2 阶 + 1 阶
思路
- 首先要想到,你第n阶的楼梯,你需要在n-1楼梯上走一步,或者是在n-2楼梯上爬俩步。相当于n阶楼梯的方法等于 n-1楼梯方法加上n-2楼梯方法
暴力法 直接递归
class Solution {
public int climbStairs(int n) {
if( n < 3) {
return n;
}
return climbStairs(n-1) + climbStairs(n+1);
}
}
复杂度分析
时间复杂度O(2^n),每一次递归都有继续增加俩个函数来继续递归,相当于二叉树一样,往下走
空间复杂的O(n),递归树的深度可以达到 n。
优化递归
上面办法中,有很多楼梯重复计算了爬上去的方法,我们使用一个数组来记录每个楼梯的方法数
class Solution {
public int climbStairs(int n) {
int memo[] = new int[n+1];
return climbStairsMemo(n, memo);
}
public int climbStairsMemo(int n, int[] memo) {
if (memo[n] > 0) { //n这个楼梯数已经计算过了,直接返回,不用重复计算
return memo[n];
}
if (n <= 3) {
memo[n] = n;
} else {
memo[n] = climbStairsMemo(n - 1, memo) + climbStairsMemo(n - 2, memo);
}
return memo[n];
}
}
复杂度分析
时间复杂度:O(N)
空间复杂度:O(N)
解法2使用数组来记住之前的方法次数
创建一个数组,数组下标 可以看做几阶楼梯的方法
对于下标里数组的值为 方法个数。
创建n+1长度数组
class Solution {
public int climbStairs(int n) {
if(n <= 3) return n;
int[] a = new int[n+1];
a[0] = 0;
a[1] = 1;
a[2] = 2;
a[3] = 3;
int i = 4;
while(i <= n) {
a[i] = a[i-1] + a[i-2];
i++;
}
return a[n];
}
}
时间复杂度O(n),只需要循环到n的楼梯。
空间复杂度O(n),创建一个大小为n+1的数组
使用俩个变量,记住n-1 和n-2的方法次数
class Solution {
public int climbStairs(int n) {
if( n < 3) {
return n;
}
int a = 1, b = 2, c = 0;
for (int i = 3; i <= n; i++) {
c = a + b;
a = b;
b = c;
}
return c;
}
}
时间复杂度:O(n),单循环到 n,需要计算第 n个
空间复杂度:O(1),使用常量级空间。