1. 程式人生 > >python實現斐波拉契數列

python實現斐波拉契數列

第一種

# 斐波拉契數列


def fib(max):
    n, a, b = 0, 1, 1
    while n < max:
        print(b)
        a, b = b, a+b
        n = n+1
    return 'done'

fib(5)

結果為

1
1
2
3
5

第二改成生成器模式
仔細觀察,可以看出,fib函式實際上是定義了斐波拉契數列的推算規則,可以從第一個元素開始,推算出後續任意的元素,這種邏輯其實非常類似generator。

也就是說,上面的函式和generator僅一步之遙。要把fib函式變成generator,只需要把print(b)改為yield b就可以了:

def fib(max):
    n, a, b = 0, 1, 1
    while n < max:
        yield b
        a, b = b, a+b
        n = n+1
    return 'done'

f= fib(5)
print(f)
for i in f:
    print(i)

結果為

<generator object fib at 0x00000170CFFD6A98>
1
2
3
5
8