【蓝桥杯】【动态规划】母牛的故事

题目描述

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

输入

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

输出

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

样例输入

2
4
5
0

样例输出

2
4
6

分析

python单纯使用递归会超时，因此使用动态规划。

dp[i]表示第i年的母牛数量

初始化dp=[0,1,2,3,4]

dp[i] =dp[i-1] +dp[i-3] 表示当前母牛数量等于上一年的数量加上具有生育能力的母牛的数量

def cow(year):
    dp=[0,1,2,3,4]

    for i in range(5,100):
        a =dp[i-1] +dp[i-3]
        dp.append(a)

    return dp[year]
while True:
    try:
       n=int(input())
       if n==0:
          break
          print(cow(n))
       catch:
          break