爬楼梯
第五题
前言
要想学好计算机算法至关重要。
一、leetcode算法第五题
假设你正在爬楼梯。需要 n 阶你才能到达楼顶。
每次你可以爬 1 或 2 个台阶。你有多少种不同的方法可以爬到楼顶呢?
示例 1:
输入:n = 2
输出:2
解释:有两种方法可以爬到楼顶。
1 阶 + 1 阶
2 阶
示例 2:
输入:n = 3
输出:3
解释:有三种方法可以爬到楼顶。
1 阶 + 1 阶 + 1 阶
1 阶 + 2 阶
2 阶 + 1 阶
提示:
1 <= n <= 45
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/climbing-stairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解题方法:
思路
如果我们可以知道这个第n个总数是(第n-1个+第n-2个的和)类似于斐波那契去数列。因此可以用解斐波那契的方法解这个题。
1.方法一:递归
思想:
首先创建一个数组用来保存之前已经得到的值。这样等再次计算这个值时,可以直接返回,不用在进行递归。
代码:
class Solution {
public int climbStairs(int n) {
int memo[] = new int[n+1]; //保存爬n个台阶的解的个数,从1开始
return climbStartsMemo(n,memo); //解爬楼梯。
}
private int climbStartsMemo(int n, int[] memo) {
if(memo[n]>0){ //如果爬n个台阶有解那么直接返回这个解
return memo[n];
}
if(n==1){ //如果只爬1个台阶它的解为1
memo[n]=1;
}else if(n==2){ //如果只爬2个台阶它的解为2
memo[n]=2;
}else{
//如果大于2个台阶那么他的解是它前两个解之和。
memo[n]=climbStartsMemo(n-1,memo)+climbStartsMemo(n-2,memo);
}
return memo[n];
}
}
时间复杂度:O(n)
空间复杂度:O(n)
2.方法二:动态规划
思想:
和上一个方法相似,不同的是:这个方法是先将之前的数计算出来并保存然后进行循环。
代码
class Solution {
public int climbStairs(int n) {
if(n==1){
return 1;
}
int[] nums = new int[n+1]; //保存值
nums[1] = 1;
nums[2] = 2;
for(int i=3;i<n+1;i++){
nums[i] = nums[i-1]+nums[i-2];
}
return nums[n];
}
}
时间复杂度:O(n)
空间复杂度:O(n)
2.方法三:
思想:
第三个方法是保存三个数,
一个保存当前的n个台阶的解,
一个保存当前的n-1个台阶的解
一个保存当前的n-2个台阶的解
代码
class Solution {
public int climbStairs(int n) {
if(n==1){
return 1;
}
int first = 1; //n-2个台阶的解
int second = 2; //n-1个台阶的解
for(int i=3;i<=n;i++){
int third = first + second; //n个台阶的解
first = second;
second = third;
}
return second;
}
}
时间复杂度:O(n)
空间复杂度:O(1)