面試題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)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章