PTA习题-python 7-8 母牛问题分数

设想一头小母牛从第4个年头开始每年生育一头小母牛。现有一头小母牛，按照此设想，第n年时有多少头母牛？

输入格式:

测试数据有多组，处理到文件尾。每组测试输入一个正整数n（1≤n≤40）。

输出格式:

对于每组测试，输出第n年时的母牛总数。

输入样例:

5
8
15

输出样例:

3
9
129

出处:

ZJUTOJ 1182

分析

1. 用一个列表存储各个年度的母牛数量。
2. 第n年的母牛数量 = 第n-1年的数量 + 新增数量。新增数量=3年前的数量。

代码实现

def a(n):
    if n<4:
        return 1
    else:
        return a(n-1)+a(n-3)
try:
    while 1:
        n = int(input())
        print(a(n))
except:
    pass