python：算法竞赛入门之一

belldeep

已于 2024-04-20 14:16:04 修改

阅读量398

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python 算法竞赛 Fibonacci lua

于 2024-04-08 21:17:32 首次发布

本文链接：https://blog.youkuaiyun.com/belldeep/article/details/137524108

计算斐波那契数列（Fibonacci sequence），不受长整型位数限制。

编写 fibonacci.py 如下

# -*- coding: utf-8 -*-
""" 计算 斐波那契数列（Fibonacci sequence）"""
import sys
from datetime import datetime

# 递归函数定义
def fib(n):
    if n <=0: return 0
    if n ==1: return 1
    if n ==2: return 1
    return fib(n-1) + fib(n-2)

# 将递归改为迭代可以把效能提升不少
def fib1(n):
    x,y = 0,1
    while n>0:
        x,y = y,x+y
        n = n-1
    return x

# 通过将计算结果保存到 dict中，后面计算时可以取用，称为备忘方式
known = {0:0, 1:1, 2:1}

def fib2(n):
    if n <=0: return 0
    if n in known: return known[n]
    res = fib2(n-1) + fib2(n-2)
    known[n] = res
    return res

# main()
if len(sys.argv) ==2:
    n = int(sys.argv[1])
else:
    print(' usage: python fibonacci.py n ')
    sys.exit(1)

if n < 3:
    print(' input n >= 3 ')
else:
    print(datetime.now())
    print('fib1(%d)= %d' % (n, fib1(n)))
    print(datetime.now())
    print('fib2(%d)= %d' % (n, fib2(n)))
    print(datetime.no