题目
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
P.S. 题目来源于leetcode,这是我的题解
解题思路
是面试题10-1-Fibonacci数列和70.爬楼梯的结合。
走到第n阶梯的总的方法数:因为只能一次爬1步或2步,则走到第N阶梯->从第N-1阶梯走一步到达 或 从第N-2阶梯到达。则总的方法数为:F[n]=F[n-1]+F[n-2]
。为Fibonaci数。
初始状态:F[0]=0,F[1]=1
借助两个辅助变量f,g
实现爬楼梯:
-f-->g
- f-->g
同时我们也像面试题10-I一样对它去下模。
时间复杂度O(n)
空间复杂度O(1)
代码
class Solution:
def numWays(self, n: int) -> int:
#DP:与70.爬楼梯以及面试题10-II相似,就是需要取模
dp = [0]*(n+1)
if n == 0: return 1
if n == 1: return 1
dp[0] = dp[1] = 1
for i in range(2,n+1):
dp[i] = dp[i-1]+dp[i-2]
if dp[i] >10**9+7: dp[i] -= (10**9+7)
return dp[n]