面試題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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章