写个程序:一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n级的台阶总共有多少种跳法。 答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
时间: 2024-02-24 19:59:19 浏览: 125
剑指Offer:一只青蛙一次可以跳上1级台阶,也可以跳上2级。求该青蛙跳上一个n级的台阶总共有多少种跳法
5星 · 资源好评率100%
好的,以下是 Python 代码实现:
```python
def jumpFloor(n):
if n <= 2:
return n
a, b = 1, 2
for i in range(3, n+1):
a, b = b, (a + b) % 1000000007
return b % 1000000007
n = int(input())
print(jumpFloor(n))
```
在这个程序中,我们使用了动态规划的思想来解决这个问题。当台阶数为 1 或 2 时,跳法分别为 1 和 2,直接返回即可。当台阶数大于 2 时,我们使用两个变量 a 和 b 来保存前两个状态的跳法数,然后从 3 开始迭代计算每个状态的跳法数,并用 b 来保存当前状态的跳法数。最后返回 b 对 1000000007 取模的结果即可。
需要注意的是,由于题目要求对结果取模,因此在计算过程中要不断取模,否则可能会出现整型溢出的情况。同时,如果最终结果为 1000000008,需要返回 1。
阅读全文