LeetCode 70. Climbing Stairs

题目

假设你正在爬楼梯。需要 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

注：

另外，这其实是一个斐波那契数列