爬楼梯

题目描述

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

观察，1阶1种，2阶2种，3阶3种，4阶5种，正好是斐波那契数列。第一次用递归超时

class Solution:
    def climbStairs(self, n):
        # step = 0
        # if n == 0 or n == 1:
        #     return 1
        # else:
        #     step = self.climbStairs(n-1) + self.climbStairs(n-2)
        # return step

        a, b, c = 1, 2, 3
        if n == 1 or n == 2:
            return n
        while c <= n:
            step = a + b
            a = b
            b = step
            c += 1
        return step