剑指 Offer 10- I. 斐波那契数列
说明
写一个函数,输入 n ,求斐波那契(Fibonacci)数列的第 n 项。斐波那契数列的定义如下:
F(0) = 0, F(1) = 1
F(N) = F(N - 1) + F(N - 2), 其中 N > 1.
斐波那契数列由 0 和 1 开始,之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例
输入:n = 2
输出:1
输入:n = 5
输出:5
代码实现
class Solution(object):
def fib(self, n):
"""
:type n: int
:rtype: int
"""
f0 = 0
f1 = 1
fn = 0
if n == 0:
return f0
if n == 1:
return f1
for _ in range(2, n+1):
fn = f0 + f1
f0 = f1
f1 = fn
return fn % 1000000007