写一个函数,输入 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。
我们在学习c语言的时候肯定接触过这个题目,没有也没关系,用递归做很简单。但是在leetcode 上用最开始的那种递归方法就会显示时间超出限制。我们用循环做也可以达到一样的效果,斐波那契数列:0,1,1,2,3,5,8,13…等等。
两个变量的循环就可以达到这种效果,首先a = 0, b = 1,把他们相加的结果复制给b,再把b的结果赋值给a,再进行下一次循环就ok了。
接下来看代码:
class Solution:
def fib(self, n: int) -> int:
a,b = 0, 1
for _ in range(n):
a, b = b, a+b
return a % 1000000007
这里取余可能是为了判题方便。不让结果过大,数字超过一定大小就重头再来。
复制语句写在一行可以节省一个变量,写成多行的形式也是可以的,见下代码:
class Solution:
def fib(self, n: int) -> int:
a = 0
b = 1
for _ in range(n):
t = a + b
a = b
b = t
return a % 1000000007
那么问题来了,为什么返回的是a呢?题目说了,N是>1 的,也就是说N最小就是2,此时返回的结果应该是1,此时a = 1, b = 2,所以应该选择a作为函数的返回值。