python基础第二天（母牛的故事）

题目：

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

        输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。n=0表示输入数据的结束，不做处理。

代码思维：

        分为两步
                第一步按输入格式读取输入的数字：

                        先新建一个列表，再用循环遍历

                                

data = []  # 存储输入数据的列表

# 读取输入数据
while True:
    n = int(input())
    if n == 0:
        break
    data.append(n)

         第二步计算数量

                我想的方法就是数学的方法：

                        找规律累加n+（n-4）+(n-5)+.....+1求和公式

def f(num,a=0,b=0,c=0,d=1):
    if num <= 1:
        print(a + b + c + d)
        return
    d += c
    f(num - 1, d, a, b, d)
 

while True:
    n = int(input())
    if n == 0:
        break
    f(n)
    

但是答案显示报错，通过gpt得知（根据这个问题的描述，母牛的增长是跟斐波那契数列有关的，而不是简单的线性关系。）参考大佬题解如图：

可以得出用 a b c d 四个来动态循环，只要到了4岁就会瞬间生下一头牛，所以刚成年的3岁，要刚成年的曾经的3岁数+成年总数 才是1岁的数量，所以需要d += c

代码

def f(num,a=0 ,b=0,c=0,d=1):
    if num <= 1:
        print(a+b+c+d)
        return
    d += c
    f(num-1,d,a,b,d)

while True:
    n = int(input())
    if n == 0:
        break
    f(n)

或者：在第n年，母牛的数量等于第n-1年的母牛数量加上第n-3年的新生小母牛数量。我们使用一个dp列表来存储每年的母牛数量。首先，我们将前三年的数量固定为1、2和3。然后，通过一个循环从第四年开始计算，将前一年的母牛数量和三年前的新生小母牛数量相加。

data = []  

while True:
    n = int(input())
    if n == 0:
        break
    data.append(n)
    
def cow_count(year):
    if year < 4:
        return year
    dp = [0] * (year + 1)
    dp[1] = 1
    dp[2] = 2
    dp[3] = 3
    for i in range(4, year + 1):
        dp[i] = dp[i-1] + dp[i-3]
    return dp[year]

for n in data :
    print(cow_count(n))