设想一头小母牛从第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