问题描述来源leetcode
70爬楼梯
假设你正在爬楼梯。需要 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 阶
提示: 1 <= n <= 45
思路:
第一阶1
第二阶可以从前一阶直接登来,或者从前2阶直接来
第三阶可以从前一阶来或者从前2阶来
于是假设现在是第n阶,那么可以从第n - 1阶或者第n-2阶直接登来。
于是这个就是规律了。
就是第n阶有f(n)个方法
那么
f(1)=1
f(2)=2
f(3)=f(2)+f(1)
...
f(n)=f(n-1)+f(n-2)
于是可以了。
题解1:
class Solution {
public int climbStairs(int n) {
if (n <= 2) return n;
int[] arr = new int[n];
arr[0] = 1;
arr[1] = 2;
for (int i = 2; i < arr.length; i++) {
arr[i] = arr[i -1] + arr[i-2];
}
return arr[n-1];
}
}
这个题解使用数组来记录前面的每一个阶数登上的使用方法,效率高一些。
因为只用到前面的俩个变量,因此还可以这样:
题解2
class Solution {
public int climbStairs(int n) {
if (n <= 2) return n;
int a = 1;
int b = 2;
for (int i = 3; i <= n; i++) {
int tmp = a + b;
a = b;
b = tmp;
}
return b;
}
}
如果是想用那些递归的思路,不考虑效率、不考虑是否会超时,那么可以这么写:
class Solution {
public int climbStairs(int n) {
if (n <= 2) return n;
return climbStairs(n - 1) + climbStairs(n - 2);
}
}
但是肯定是会超时了:
也就最后一个测试案例没通过。这就是代码对,但是超时:
还有一种可以用回溯的,也是会耗很长时间的,写起来稍微复杂些,等后面有时间再写吧