7-1 h0191. 母牛的故事

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

输入格式:

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

输出格式:

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

输入样例:

2
4
5
0

输出样例:

2
4
6

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int x;
    int a[55]={0},b[55]={0};
    for(int i=1;i<=55;i++){
        cin>>x;
        if(x==0)break;
        b[i]=x;
    }
    a[1]=1,a[2]=2,a[3]=3,a[4]=4;
    for(int i=4;i<=55;i++){
        a[i]=a[i-1]+a[i-3];
    }
    for(int i=1;i<=55;i++){
        if(b[i]==0)break;
        cout<<a[b[i]]<<endl;
    }
}