剑指Offer 10- I. 斐波那契数列
2022年3月04日
一、问题描述
写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项(即 F(N))。斐波那契数列的定义如下:
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:1
示例 2:
输入:n = 5
输出:5
提示:
0 <= n <= 100
二、问题分析
- 试过递归求解,但是由于指数级的时间复杂度超时了,所以考虑循环求解。
- 这里考虑从0开始循环构造斐波那契数列,通过循环将前两项与前一项相加,不断循环直到第n项。
三、解题代码
class Solution(object):
def fib(self, n):
"""
:type n: int
:rtype: int
"""
MOD = 10 ** 9 + 7
if n < 2:
return n
p, q, r = 0, 0, 1
for i in range(2, n + 1):
p = q
q = r
r = (p + q) % MOD
return r
总结
为了时间复杂度使用循环(线性时间复杂度)替换递归(指数时间复杂度)即可。