《剑指offer》刷题系列——（三十三）青蛙跳台阶问题

题目

一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。
示例：
输入：n = 2
输出：2

思路

动态规划
青蛙每次只能跳1个台阶或者2个台阶。
如果跳上 n 级台阶有 f(n)种跳法。
当最后一步跳 1级台阶时，总的跳法数为 f(n−1)种；
当最后一步跳 2级台阶时，总的跳法数为 f(n−2)种。

这道题可以转化为 斐波那契数列 问题，只需将初始值改为 f(0)=1，f(1)=1

代码

class Solution:
    def numWays(self, n: int) -> int:
        if n== 0: return 1
        if n== 1: return 1
        a,b = 1,1
        for _ in range(n-1):
            a,b = b,a+b
        return b%1000000007

复杂度

时间复杂度 O(N)
空间复杂度 O(1)