题解一:
动态规划法
爬楼梯问题可以分而治之,被分解为包含最优子结构的子问题。
分析:
上第i阶的方法有:
1.在第(i-1)阶后向上爬一阶
2.在第(i-2)阶后向上爬二阶
所以到达第i阶的方法总数是到第(i-1)阶和第(i-2)阶的方法数之和。
令dp[i]表示能到达第i阶的方法总数,则有:
dp[i]=dp[i-1]+dp[i-2]
伪代码:
输入:输入台阶总数整数n
输出:爬到楼顶的总方法
1.如果n=1,返回1
2.创建长度为n+1的空数组dp,其dp[i]表示能到达第i阶的方法总数
3.初始化:dp[1]=1,dp[2]=2
4.对于索引i从3到n,建立循环:dp[i]=dp[i-1]+dp[i-2]
5.返回dp[n]
时间复杂度:O(n),一个循环
空间复杂度:O(n),dp数组占用了n的空间
java代码
class Solution {
public int climbStairs(int n) {
if(n == 1) {
return 1;
}
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];
}
}
python
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
import numpy as np
if n==1:
return 1
dp=np.zeros(n+1,dtype=int)
dp[1]=1
dp[2]=2
for i in range(3,n+1):
dp[i]=dp[i-1]+dp[i-2]
return dp[n]
题解二:
斐波那契数列详解
斐波那契数列(Fibonacci sequence),又称黄金分割数列
斐波那契数列指的是这样一个数列:从第三项开始,每一项都等于前两项之和。
观察数学规律,可知本题是斐波那契数列,那么用斐波那契数列的公式即可解决问题,公式如下:
F n = 1 / 5 [ ( 1 + 5 2 ) n − ( 1 − 5 2 ) n ] F_n=1/\sqrt{5} [( \frac{1+\sqrt{5} }{2}) ^n-( \frac{1-\sqrt{5} }{2}) ^n] Fn=1/5[(21+5)n−(21−5)n]
时间复杂度:O(logn)
空间复杂度:O(1)
java
class Solution {
public int climbStairs(int n) {
double sqrt5=Math.sqrt(5);
double fibn=Math.pow((1+sqrt5)/2,n+1)-Math.pow((1-sqrt5)/2,n+1);
//pow(x, n) ,即计算 x 的 n 次幂函数
return (int)(fibn/sqrt5);
}
}
python
class Solution(object):
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
import math
sqrt5=math.sqrt(5)
fibn=pow((1+sqrt5)/2,n+1)-pow((1-sqrt5)/2,n+1)
return int(fibn/sqrt5)