题目:
假设你正在爬楼梯。需要 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 阶
算法实现:
方法一:递归
class Solution:
def climbStairs(self, n: int) -> int:
a = 0
def fun(n):
nonlocal a
if n == 1 or n == 0:
a += 1
else:
fun(n - 1)
fun(n - 2)
fun(n)
return a
结果:
由于重复步骤太多,导致超出时间限制
方法二:记忆化递归
利用字典保存每次递归的结果,再下次递归时直接调用结果
class Solution:
def climbStairs(self, n: int) -> int:
m = {1: 1, 0: 1}
def helper(n):
if n in m:
return m[n]
a = helper(n - 1) + helper(n - 2)
m.setdefault(n, a)
return a
return helper(n)
结果:
方法三:排列组合
class Solution:
def climbStairs(self, n: int) -> int:
i = 0
a = 0
while i <= n:
a += int(math.factorial(n)/(math.factorial(i) * math.factorial(n - i)))
i += 1
n -= 1
return a