动态规划母牛的故事

解题思路:

注意事项:

参考代码:

N = []       #   所有要求的年份（题意是求第n年初，所有之后索引值要减一）
while 1:
    n = int(input())
    if n == 0:
        break
    else:
        N.append(n)

cow = [0,0,0,1]    # 分别代表age=1，age=2，age=3，age>=4的母牛数量（当前为第0年末，之后一年一年的更新）
num = [sum(cow)]   # 索引0即第0年末的母牛总数

def update():      #  状态转移方程
    temp = cow[3]
    cow[1],cow[2],cow[3] = cow[0],cow[1],cow[2]
    cow[3] += temp
    cow[0] = cow[3]

for _ in range(max(N)-1):
    update()
    num.append(sum(cow))

for i in N:
    print(num[i-1])