题目
假设你正在爬楼梯。需要 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 阶
解法
图解思路:
所以我们可以用「滚动数组思想」把空间复杂度优化成 O(1)。
class Solution {
public:
int climbStairs(int n) {
// dp: f(x) = f(x-1) + f(x-2)
// 滚动动态数组思想将空间复杂度从O(n)变为O(1)
int p = 0, q = 0, r = 1;
for(int i = 1; i <= n; i++){
p = q;
q = r;
r = p + q;
}
return r;
}
};
笔记
法二(普通人思路)
递归error : 在n = 45的时候超时。
class Solution {
public:
int f(int x){
if(x == 0 || x == 1) return 1;
return f(x-1) + f(x-2);
}
int climbStairs(int n) {
int ans = f(n);
return ans;
}
};
可以修改为以下写法:
class Solution {
public:
int climbStairs(int n) {
int* dp = new int[n+2]();
//dp[0] = 0; //不管怎样,数组下标指针肯定是从0开始的,所以要考虑0.有0个台阶,不需要爬,所以没有方法数(但从斐波那契角度,dp[0]=1)
dp[1] = 1;
//2阶台阶,有两种方式(1+1, 2), 因为题目设定n是正整数,所以n最小是1,此时如果定义dp的长度是int[n + 1],则length=2
//而dp[2]实际对应的是第三个元素,超出length了,所以定义new int[n + 2]更合理
dp[2] = 2;
for(int i = 3; i <= n ; i++){
dp[i] = dp[i-1] + dp[i-2];
}
return dp[n];
}
};