PTA Python 6-2 兔子繁殖

兔子繁殖问题。假设有一对新生的兔子，从第三个月开始他们每个月月初都生一对兔子，新生的兔子从第三个月月初开始又每个月又生一对兔子。按此规律，并假定兔子没有死亡，n(n<=20)个月月末共有多少对兔子？

函数接口定义：

函数接口： rabit(n)，n为第几个月

裁判测试程序样例：

n=int(input()) print(rabit(n))

输入样例1：

在这里给出一组输入。例如：

3

输出样例1：

在这里给出相应的输出。例如：

2

输入样例2：

在这里给出一组输入。例如：

5

输出样例2：

在这里给出相应的输出。例如：

5

答案:

 

def rabit(n):
    a2 = 1
    a1 = 0
    a = 0
    for x in range(n-1):
        x2 = a2
        x1 = a1
        x = a
        a2 = x+x1
        a1 = x2
        a += x1

    return a1+a2+a

​