剑指 Offer 10- I. 斐波那契数列、青蛙跳台阶问题（python）

斐波那契数列

题干

写一个函数，输入 n ，求斐波那契（Fibonacci）数列的第 n 项（即 F(N)）。斐波那契数列的定义如下：
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1
斐波那契数列由 0 和 1 开始，之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。
示例1：
输入：n = 2
输出：1
示例2：
输入：n = 5
输出：5
0 <= n <= 100

解题思路

斐波那契数列是典型问题，一般用循环或递归实现。F(0) = 0, F(1) = 1是递归终点

代码

class Solution:
    def fib(self, n: int) -> int:
        pre1 = 0
        pre2 = 1
        for i in range(n):
            pre1, pre2 = pre2, pre1 + pre2
        return pre1 % 1000000007     # 题干要求取余

青蛙跳台阶问题

题干

解题思路

本题很显然与斐波那契数列是同类的问题，只是初始的条件有稍微区别，可以举例最初的0个台阶、1个台阶为例得出pre1和pre2的值。
当有n级台阶时、上一步可能是跳了1级、也可能跳了2级，因此n级台阶的结果 = n-1级台阶+ n-2级台阶。依次递归循环即可

代码

class Solution:
    def numWays(self, n: int) -> int:
        pre1 = 1
        pre2 = 1
        for i in range(n):
            pre1, pre2 = pre2, pre1 + pre2
        return pre1 % (1000000007)