题目
假设你正在爬楼梯。需要 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 阶
思路
这个题非常典型,可以用递归很容易做出来,上n阶台阶可以分解为上n-1阶台阶后再上一阶和上n-2阶台阶后再上2阶,递归即可。
然而做这道题的时候当n>30左右就会提示超时,那只好以空间换时间,开辟一个字典存放阶数和对方法数。因为递归存在重复调用,所以只要求出就记录下来,之后只要查找字典就行了,能省很多时间。
有个小插曲,把s1的部分和s2的部分顺序反一下也会超时,想想也就明白了:这是因为递归造成的,n-2的话就有很多没有被算到,递归的非常乱。n-1的话每次返回再去做n-2的话就不用重新递归,直接找字典里的就可以了,相当于线性时间复杂度。
代码
注意一下steplist需要是全局的
class Solution:
def climbStairs(self, n):
"""
:type n: int
:rtype: int
"""
self.steplist = {1:1,2:2}
if n == 1:
return 1
if n == 2:
return 2
if n > 2:
s1 = self.steplist.get(n-1)
if not s1:
s1 = self.climbStairs(n-1)
self.steplist[n-1] = s1
s2 = self.steplist.get(n-2)
if not s2:
s2 = self.climbStairs(n-2)
self.steplist[n-2] = s2
t = s1 + s2
self.steplist[n] = t
return t
注:
另外,这其实是一个斐波那契数列