统计每个月兔子的总数（华为机试练习）

题目描述

有一只兔子，从出生后第3个月起每个月都生一只兔子，小兔子长到第三个月后每个月又生一只兔子，假如兔子都不死，问每个月的兔子总数为多少？

代码

while True:
    try:
        month=int(input())
        a,b=1,0
        for i in range(month):
            a,b=b,a+b
        print(b)
    except:
        break

测试