题目
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
1 <= n <= 45
示例
输入:n = 2
输出:2
解释:有两种方法可以爬到楼顶。
1. 1 阶 + 1 阶
2. 2 阶
输入:n = 3
输出:3
解释:有三种方法可以爬到楼顶。
1. 1 阶 + 1 阶 + 1 阶
2. 1 阶 + 2 阶
3. 2 阶 + 1 阶
解决方案
方法一:动态规划
每次只能走一步或者两步,那就是说,n个台阶需要走多少次,
等于 先走一步剩下的(n-1)个台阶需要的步数 加上 先走两步剩下的(n-2)个台阶需要的步数。
后来去查了一下,DP(n) = DP(n-1) + DP(n-2),这是典型的动态规划。
class Solution {
public int climbStairs(int n) {
if(n < 0) {
return 0;
}
if(n == 0) {
return 1;
}
return climbStairs(n - 1) + climbStairs(n - 2);
}
}
虽然这样写,当n为44时就超时了,但掌握这种思想还是有帮助的。
方法二:动态数组
如果将每个台阶数对应的步数方法数放在一个数组中,根据动态规划的思想可以得到元素之间的关系:nums[i] = nums[i-2]+nums[i-1],因为这个数组的前几位是可以手推的,所以不停遍历累加几就好。
class Solution {
public int climbStairs(int n) {
// p、q、r分别对应i-2, i-1, i
// 初始值 p对应n=-1 q对应n=0 r对应n=1
int p = 0, q = 0, r = 1;
for (int i = 1; i <= n; i++) {
p = q;
q = r;
r = p + q;
}
return r;
}
}