假设你正在爬楼梯。需要 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 阶
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/climbing-stairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路一:记忆化递归:根据规则知道:f(n) = f(n-1) + f(n-2), f(0) = 0
class Solution:
def climbStairs(self, n: int) -> int:
self.dict = {}
self.ans = 0
return self.dfs(n)
def dfs(self, n):
if n < 0:
return 0
if n == 0:
self.ans += 1
self.dict[n] = 1
return 1
if n not in self.dict:
self.dict[n] = self.dfs(n - 1) + self.dfs(n - 2)
return self.dict[n]
思路二:动态规划,其实跟递推差不多思路,只是这个是正向求解:f(n) = f(n-1) + f(n-2)
class Solution:
def climbStairs(self, n: int) -> int:
dict = {}
dict[1] = 1
dict[2] = 2
for i in range(3, n + 1):
dict[i] = dict[i-1] + dict[i-2]
return dict[n]
思路三:针对思路二进行优化,去掉dict记录,可以想到斐波那契数列,见下面代码。
class Solution:
def climbStairs(self, n: int) -> int:
if n == 1:
return 1
d1 = 1
d2 = 2
for i in range(3, n + 1):
d = d1 + d2
d1 = d2
d2 = d
return d2
方法五:Binets 方法
方法六:斐波那契公式
import math
class Solution:
def climbStairs(self, n: int) -> int:
ans = int((1/math.sqrt(5)) * ((0.5 + math.sqrt(5)/2)**(n+1) - (0.5 - math.sqrt(5)/2)**(n+1)))
return ans