面试题10- II. 青蛙跳台阶问题

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

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

思路:

青蛙第一次跳的时候有两种跳法:跳一阶,跳两阶。
如果跳一阶:剩下的跳法等于后面n-1阶的跳法数。
如果跳两阶:剩下的跳法等于后面n-2阶的跳法数。
即:f(n) = f(n-1) + f(n-2)

class Solution:
    def numWays(self, n: int) -> int:
        a, b = 1, 1 #这两种跳法初始化各自消耗一次
        for _ in range(n):
            a, b = b, a + b
        return a % 1000000007
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章