题目链接
https://leetcode-cn.com/problems/climbing-stairs/
描述
假设你正在爬楼梯。需要 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 阶
代码
简单的动态规划思想,当前的楼梯层只能由前一层或者前两层达到,因此
class Solution {
public int climbStairs(int n) {
if (n <= 2) {
return n;
}
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];
}
}
当前的楼梯层数到达的方法数只和前两层的层数有关,因此,可以利用两个常量,不再使用数组
class Solution {
public int climbStairs(int n) {
int prev1 = 0;
int prev2 = 1;
int cur = 0;
for (int i = 1; i <= n; i++) {
cur = prev1 + prev2;
prev1 = prev2;
prev2 = cur;
}
return cur;
}
}