一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 0
输出:1
这里要记录一下小青蛙是怎么转换成斐波那契的问题的
设跳上 n nn 级台阶有 f ( n ) f(n)f(n) 种跳法。在所有跳法中,青蛙的最后一步只有两种情况: 跳上 1 11 级或 2 22 级台阶。
- 当跳上 1 11 级台阶时: 之前共跳了 n − 1 n-1n−1 个台阶,此情况有 f ( n − 1 ) f(n-1)f(n−1) 种跳法;
- 当跳上 2 22 级台阶时: 之前共跳了 n − 2 n-2n−2 个台阶,此情况有 f ( n − 2 ) f(n-2)f(n−2) 种跳法。
f ( n ) f(n)f(n) 为以上两种情况的和,即 f ( n ) = f ( n − 1 ) + f ( n − 2 ) f(n)=f(n-1)+f(n-2)f(n)=f(n−1)+f(n−2),则该递推性质为斐波那契数列。本题可转化为 求斐波那契数列第 n nn 项的值,与 求斐波那契数列 等价,唯一的不同在于起始数字不同。
- 青蛙跳台阶问题: f ( 0 ) = 1 f(0)=1f(0)=1 , f ( 1 ) = 1 f(1)=1f(1)=1 , f ( 2 ) = 2 f(2)=2f(2)=2 ;
- 斐波那契数列问题: f ( 0 ) = 0 f(0)=0f(0)=0 , f ( 1 ) = 1 f(1)=1f(1)=1 , f ( 2 ) = 1 f(2)=1f(2)=1 。
class Solution:
def numWays(self, n: int) -> int:
a, b = 1, 1
for _ in range(n):
sum = (a + b) % 1000000007
a, b = b, sum
result = a % 1000000007
return result
版权声明:本文为Wolf_xujie原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。