面试题10 II, 青蛙跳台阶问题(Python3) 递归

一只青蛙一次可以跳上1级台阶，也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

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

示例 1：

输入：n = 2
输出：2
示例 2：

输入：n = 7
输出：21

题目分析

这也是一个具有递推性质的题。设第n阶有f(n)中跳法，则递推关系为：

f(n)=f(n-1)+f(n-2)

可以理解成f(n)中跳法中每一种跳法的最后一跳要么是1阶，要么是2阶。如果是1阶，则前(n-1)阶对应的跳法是(n-1)种；如果是两阶，则前(n-2)阶对应的跳法是f(n-2)种。

其中，f(1)=1;f(2)=2.

这本质上和斐波那契数列问题是一样的。

 

代码

class Solution:
    def numWays(self, n: int) -> int:
        a = 1
        b = 1
        if n == 0:
            return 1

        if n == 1:
            return 1

        for i in range(n-1):
            a,b = b, a+b
        return b%1000000007

还可以对代码做如下简化：

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

具体含义见昨天的博客：面试题10 I. 斐波那契数列 (Python3)