Python 计算斐波那契数列

今天来练习使用 Python 计算斐波那契数列。由于斐波那契数列第一项的值为 1，所以下面的代码也是以第一项为 1 开始计算。

计算斐波那契数列的第n 项的值

方法 1

由于 Python 解释器默认的最大递归深度为 1000，所以在不更改解释器的最大递归深度的情况下，这种方式只能计算斐波那契数列的前一千个值，而且由于 Python 未对尾递归进行优化，所以这种方式的效率特别低，计算 100 个数值已经让人没心情等下去了。

def fib1(n):
    if not isinstance(n, int) or n < 1:
        return None
    if n in (1, 2):
        return 1
    while n > 2:
        return fib1(n - 1) + fib1(n - 2)

方法 2

这种方式的效率要比第一种方式的效率高太多太多

def fib2(n):
    if not isinstance(n, int) or n <1:
        return None
    a, b, i = 0, 1, 1
    while i < n:
        a, b = b, a + b
        i += 1
    return b

计算斐波那契数列的前 n 项值

由于上面只计算了斐波那契数列的第 n 项的值，没办法打印所有前 n 项的值，所以我们需要把前 n 项的值返回。

方法 1

使用列表来存储前 n 项的值：

def fib3(n):
    if not isinstance(n, int) or n < 1:
        return None
    lib_seq = [1, 1]
    if n <= 2:
        return lib_seq[: n - 1]
    for i in range(n - 2):
        lib_seq.append(lib_seq[-1] + lib_seq[-2])
    return lib_seq

方法 2

由于所有的值都存在列表中，所以比较占内存，我们可以创建一个返回斐波那契数列值的生成器函数：

def fib4(n):
    if not isinstance(n, int) or n < 1:
        raise TypeError('n must be an integer greater than or equal to 1')
    a, b, i = 0, 1, 1
    while i <= n:
        yield b
        a, b = b, a + b
        i += 1

fib4 是一个生成器函数，可以使用 for 循环来迭代其中的元素：

>>> for i in fib4(10):
...     print(i)
... 
1
1
2
3
5
8
13
21
34
55

或者使用 iter 和 next 函数来手动迭代：

>>> it = iter(fib4(10))
>>> while True:
...     try:
...         print(next(it))
...     except StopIteration:
...         break
... 
1
1
2
3
5
8
13
21
34
55