题目描述
-
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
-
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
-
注意:给定 n 是一个正整数
-
分析:
- 要求 n 层的台阶的走法,由于一次走 1 或 2 个台阶,所以上到第 n 个台阶之前,一定是停留在第 n - 1 个台阶上,或者 n - 2 个台阶上。
- 如果用 f ( n ) 代表 n 个台阶的走法。
- 那么
f(n)=f(n-1)+f(n-2),f(1)=1,f(2)=2
。 - 发现这就是斐波那契数列(Fibonacci sequence)。
解法一:暴力法(递归)
class Solution {
public int climbStairs(int n) {
return climbStair(n);
}
private int climbStair(int n){
if(n==1)
return 1;
if(n==2)
return 2;
return climbStair(n-1)+climbStair(n-2);
}
}
- 运行结果超出时间限制(由于每次递归都要开辟栈空间,浪费了很多空间导致)
- 改进
- 原因就是先求 climbStairs (n-1),然后求 climbStairs(n-2) 的时候,其实很多解已经有了,但是它依旧进入了递归。优化方法就是把求出的解都存起来,后边求的时候直接使用,不用再进入递归了。叫做 memoization 技术。
- 如下代码 采用一个数组存储已经有的解
class Solution {
public int climbStairs(int n) {
int memo[] = new int[n + 1];
return climbStairs(n, memo);
}
private int climbStairs(int n, int[] memo) {
if (n==1) {
return 1;
}
if (n==2) {
return 2;
}
int n1=0;
// 判断f(n-1)是否已经计算出,如果有直接使用。如果没有,递归计算。
if (memo[n-1]==0) {
n1=climbStairs(n-1,memo);
memo[n-1]=n1;
} else {
n1=memo[n-1];
}
int n2=0;
// 判断f(n-2)是否已经计算出,如果有直接使用。如果没有,递归计算。
if (memo[n-2]==0) {
n2 = climbStairs(n-2,memo);
memo[n-2]=n2;
} else {
n2=memo[n-2];
}
return n1+n2;
}
}
- 时间复杂度:O(2^n)
解法二:动态规划(迭代)
- 初始值 f(1) 和 f(2),然后可以求出 f(3),然后求出 f(4) … 直到 f(n),一次循环
class Solution {
public int climbStairs(int n) {
if(n==1){
return 1;
}
if(n==2){
return 2;
}
int n1=1;
int n2=2;
for(int i=3;i<=n;i++){
int temp = n2;
n2 = n1+n2;
n1 = temp;
}
return n2;
}
}
- 时间复杂度:O(n)
- 空间复杂度:O(1)