题目
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
1、递归思路[O(2^n)]:
class Solution {
public int climbStairs(int n) {
if(n == 1){
return 1;
}else if(n == 2){
return 2;
}else{
return climbStairs(n-1) + climbStairs(n-2);
}
}
}
2、备忘录方法[O(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){ //大于0则证明已经求解过这个问题直接使用
return memo[n];
}
if(n == 1){
memo[n] = 1;
}else if(n == 2){
memo[n] = 2;
}else{
memo[n] = climbStairsMemo(n-1, memo) + climbStairsMemo(n-2, memo);
}
return memo[n];
}
}
动态规划[O(n)]
class Solution {
public int climbStairs(int n) {
if(n == 1){
return 1;
}
int[] dp = new int[n+1];
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i <= n; i++){
dp[i] = dp[i - 1] + dp[i - 2];
}
return dp[n];
}
}