假设你正在爬楼梯。需要 n
阶你才能到达楼顶。
每次你可以爬 1
或 2
个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2 输出:2 解释:有两种方法可以爬到楼顶。 1. 1 阶 + 1 阶 2. 2 阶
示例 2:
输入:n = 3 输出:3 解释:有三种方法可以爬到楼顶。 1. 1 阶 + 1 阶 + 1 阶 2. 1 阶 + 2 阶 3. 2 阶 + 1 阶
思路
动态规划,记录每层爬上去的方法,每层的前两层方法相加。
代码
class Solution {
public:
int climbStairs(int n) {
if(n == 1 || n == 2)
return n;
int dp[2] = {1,2};
int tmp;
for(int i=2; i<n; ++i)
{
tmp = dp[1];
dp[1] = dp[0] + dp[1];
dp[0] = tmp;
}
return dp[1];
}
};