母牛生小牛
描述
设有一头小母牛,从出生第四年起每年生一头小母牛,按此规律,第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