python100道经典例题第11题

python100道经典例题第11题：古典问题：有一对兔子，从出生后第 3 个月起每个月都生一对兔子，小兔子长到第三 个月 后每个月又生一对兔子，假如兔子都不死，问每个月的兔子总数为多少？

我一开始确实一下想不到好办法，就用笨办法，先列了前五个月的兔子数量关系

#old代表第3个月的兔子，new1代表第一个月的兔子，new2 代表第二个月的兔子
yue==1
old=0
new1=2
new2=0

yue==2
old=0
new1=0
new2=2

yue==3
old=2
new1=2
new2=0

yue==4
old=2
new1=2
new2=2

yue==5
old=4
new1=4
new2=2

简单列出来这个关系之后，就发现规律了：

前两个月兔子数量不变，从第三个月开始

nwe1赋值给nwe2；nwe2赋值给old并加上原来的old；old赋值给nwe1

（也试过想直接用for循环，不用加if，但第二个月的时候就出错了....）

最后重新整理得到如下：

yue=int(input('请输入当前月份'))
if yue==1 or yue==2:
    print('一共有2只兔子')
else :#以第2月的现状为准
    nwe1=0#第一个月的兔子
    nwe2=2#第二个月的兔子
    old=0#第三个月的兔子
    for i in range(3,yue+1):#每个月都是在前一个月的基础上得到的
        old=old+nwe2
        nwe2=nwe1
        nwe1=old
        sum=nwe1+nwe2+old
    print(sum)