蓝桥杯刷题打卡day1

问题 1004: 母牛的故事

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

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

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

样例输入
2
4
5
0
样例输出
2
4
6

#include<iostream>
using namespace std;
#define N 55
int cows[N];
int cow(int n);
int main(){
	int input[N];
	cow(N);
	int i=0,j;
	do
    {
        cin>>input[i];
    }while(input[i++] != 0);
    
	for(j=0;j<i-1;j++){
    	if(input[j]<4)
    	cout<<cows[input[j]]<<endl;
    	else 
    	cout<<cows[input[j]-1]+cows[input[j]-3]<<endl;
	}
	return 0;
}
int cow(int n){
	int i;
	for(i=1;i<55;i++){
		if(i<=4) {
			cows[i]=i;
		}
		else {
			cows[i]=cows[i-1]+cows[i-3];
		}
	}
}