一隻青蛙一次可以跳上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