PYTHON蓝桥杯——每日一练（简单题）

题目

有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。请编程实现在第n年的时候，共有多少头母牛？

输入格式

输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0 n=0表示输入数据的结束，不做处理。

输出格式

对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。

解析

 一、递归实现计算

二、用一个列表记录牛数量

三、最后输出数量

代码

def shu(n):
    if n<= 4:
        return n
    else:
        return shu(n-1) + shu(n-3)

nian = int(input())

s = []  #用来存储每一年牛的数量

while nian !=0:
    s.append(shu(nian))  #如果年份不是0就循环输入年份，和记录牛的数量
    nian = int(input())
for i in range(len(s)):  #输出每一个年份的牛数量
    print(s[i])

运行结果