Python3——青蛙跳台阶问题

一、题目描述
题目来自剑指Offer 10-II，难度简单。
一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。
示例 1

输入：n = 2
输出：2

示例 2：

输入：n = 7
输出：21

示例 3：

输入：n = 0
输出：1

提示：

0 <= n <= 100

二、题目解析
我们先来模拟下青蛙跳台阶的过程。

当台阶数n=0时，青蛙只能原地不动，只有1种方法。

当台阶数n=1时，青蛙只能从平地跳上1级台阶，只有1种方法

当台阶数n=2时，青蛙可以从平地先跳到1级台阶，再从1级台阶跳到2级台阶；青蛙也可以直接从平地跳到2级台阶。一共是2种方法。

当台阶数n=3时，青蛙可以从平地1级级地往上跳；青蛙可以先跳2级再跳1级；青蛙可以先跳1级再跳两级。一共是3种方法。

当台阶数n=4时，青蛙可以从平地1级级地往上跳；青蛙可以按照1-2-1 ； 2-1-1 ；1-1-2 ；2-2 这样的级数来跳。一共是5种方法。
…
以此类推
我们可以发现，青蛙跳的最后一步，不是1就是2。
假设现在台阶数为n级，有f(n)种跳法。青蛙最后一跳是1级，则前面n-1个台阶有f(n-1)种方法；青蛙最后一跳是2级，则前面n-2个台阶有f(n-2)中方法。这两种类别的总和即为n阶台阶的总方法数。
f(n) = f(n-1) + f(n-2)

可以发现，这个递推式很剑指Offer 10-I斐波那契数列一样，只是起始数字不一样。因此方法都是一样的，所以就不赘述了，直接上代码。

剑指Offer 10-I斐波那契数列

三、参考代码

# 定义了存储表dp
class Solution:
    def numWays(self, n: int) -> int:
        # 从n=3阶开始计算
        dp = {}
        dp[0] = 1
        dp[1] = 1
        dp[2] = 2
        for i in range(3,n+1):
            dp[i] = (dp[i-1] + dp[i-2]) % 1000000007
        return dp[n]
        
# 没有定义存储列表的方法，节省空间
class Solution:
    def numWays(self, n: int) -> int:
        if n==0 or n ==1:
            return 1
        elif n==2:
            return 2
        a = 1
        b = 2
        for i in range(3,n+1):
            t = b
            b = (a + b)%1000000007
            a = t
        return b