跳台阶 牛客网 python

描述

一只青蛙一次可以跳上1级台阶，也可以跳上2级。求该青蛙跳上一个n级的台阶总共有多少种跳法（先后次序不同算不同的结果）。

示例1

输入：2

复制返回值：2

示例2

输入：7

返回值：21

代码：

class Solution:
    def jumpFloor(self, number):
        if number == 1:
            return 1
        if  number == 2:
            return 2
        a, b = 1, 2
        for i in range(3, number + 1):
            a, b = b, a + b
        return b