题解 | #统计每个月兔子的总数#

借用第一赞思路，优化一下代码，还是把新生兔、一个月、两个月及以上的兔子的数量设为k0、k1、k2，不需要做判断，每个月给兔子升级就能获得各类兔子的数量，只有k2的兔子才会生小兔子。
n = int(input())
for i in range(n):
    if i == 0:   # 第一个月初始化数量
        k0 = 1
	k1 = 0
	k2 = 0
    if i > 0:   # 之后的每个月都更新一下各个级别的兔子数量
        k2 += k1  # k1的兔子升级到k2
        k1 = k0   # k0的兔子升级到k1
        k0 = k2   # 当月k2总数为新生兔子k0的数量
print(k0+k1+k2)