母牛的故事
有一头母牛,它每年年中生一头小母牛。每头小母牛从第四个年头开始,每年年中也生一头小母牛。请编程实现在第 nnn 年的时候(不算第 n年出生的小母牛),共有多少头母牛?
输入格式
输入包括一个整数 n(0<n<55)n(0 < n < 55)n(0<n<55)。
输出格式
输出在第 n年的时候母牛的数量。
我的代码如下:(用了迭代法,来减少时间复杂度)
n=int(input())
if n<=4:
print(n)
if n>4:
a=1
b=2
c=3
d=4
for i in range(n-4):
t=b+d
a=b
b=c
c=d
d=t
print(d)