蓝桥杯 C语言 【题目】母牛的故事

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

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

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

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

代码：
#include<stdio.h>
int main()
{
int n=0,m=0,a=1,i=0,j=0,k=0;
int num[55],sum[10];
scanf("%d",&n);
num[1]=num[2]=num[3]=num[4]=1;
for(i=5;i<55;i++){
num[i]=num[i-3]+num[i-1];
}
while(n!=0){
if(n<=4){
sum[j]=n;j++;
}
else{
sum[j]=num[n]*2+num[n-1]+num[n-2];
j++;
}
scanf("%d",&n);
}
k=j;
for(j=0;j<k;j++)printf("%d\n",sum[j]);
}