PYTHON练习题---设有一头小母牛，从出生第四年起每年生一头小母牛

描述

设有一头小母牛，从出生第四年起每年生一头小母牛，按此规律，第N年时有几头母牛？

输入
输入一个整数N

输出
第N年时母牛的数量

输入样例 1

5
输出样例 1

3
解决思路：递推法
由题目易知今年的牛的数量是去年牛的数和三年前牛的数的和
由此可得递推式为cow[i]=cow[i-1]+cow[i-3]

n=int(input())
cow=[1,1,1,1,2,3]
if n>5:
    for i in range(6,n+1):
        r=cow[i-1]+cow[i-3]
        cow.append(r)
    print(r)
else:
  for i in range(len(cow)):
      print(cow[n])
      break