一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
思路:动态规划
注意:temp=a; a=b; b=temp+b;return b等价于a, b = b, a+b; return a
class Solution:
def numWays(self, n: int) -> int:
# f(0)=f(1)=1 f(n)=f(n-1)+f(n-2)
a = 1
b = 1
for _ in range(2, n+1):
temp = a
a = b
b = temp + b
return b%1000000007