一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
思路:假设青蛙跳的台阶是n阶,那我们需要分情况考虑最后一跳青蛙跳的是一个台阶还是两个台阶,这是解题的关键。假设跳上n级台阶有f(n)种跳法,那如果最后跳一个台阶,前面就有n-1个台阶,有f(n-1)种跳法,如果最后跳两个个台阶,前面就有n-2个台阶,有f(n-2)种跳法,所以f(n)=f(n-1)+f(n-2),这样就又转化为斐波那契数列,只不过f(0)=1,f(1)=1
代码:
class Solution:
def numWays(self, n: int) -> int:
if n<=1:
return 1
else:
q=0
p=1
r=1#因为下面即使是2,也会循环一次
for i in range(2,n+1):
q=p
p=r
r=(p+q)%1000000007
return r