链接
https://leetcode-cn.com/problems/climbing-stairs/
耗时
解题:5 min
题解:12 min
题意
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
注意:给定 n 是一个正整数。
思路
与 剑指 Offer 10- II. 青蛙跳台阶问题 基本相同。
状态 dp[i] 可以从 dp[i-1] 跳一个台阶 转移得到,也可以从 dp[i-2] 跳两个台阶 转移得到,所以状态转移方程如下:
d
p
[
i
]
=
d
p
[
i
−
1
]
+
d
p
[
i
−
2
]
dp[i] = dp[i-1] + dp[i-2]
dp[i]=dp[i−1]+dp[i−2]
时间复杂度: O ( n ) O(n) O(n)
AC代码
class Solution {
public:
int climbStairs(int n) {
if(n < 3) return max(1, n);
vector<int> dp(n+2);
dp[1] = 1;
dp[2] = 2;
for(int i = 3; i <= n; ++i) {
dp[i] = dp[i-1] + dp[i-2];
}
return dp[n];
}
};
滚动更新(节省空间)
class Solution {
public:
int climbStairs(int n) {
if(n < 3) return max(1, n);
int pre = 1, post = 2;
for(int i = 3; i <= n; ++i) {
int tmp = post;
post = pre+post;
pre = tmp;
}
return post;
}
};