杭电oj-2018 母牛的故事 C++

Problem Description

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

Input

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

Output

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

Sample Input

2
4
5
0

Sample Output

2
4
6

#include <iostream>
//#include <string>
using namespace std;
const int  max_size=101;

int cow(int n){
int a1[max_size]={0};  //第i年活一年的牛的头数
int a2[max_size]={0};  //第i年活二年的牛的头数
int a3[max_size]={0};  //第i年活三年的牛的头数
int a4[max_size]={0};  //第i年活>=四年的牛的头数

a4[1]=1;  //最开始的那头母牛

for(int i=2;i<=n;i++)
{
   a1[i]=a4[i-1]+a3[i-1];
   a2[i]=a1[i-1];
   a3[i]=a2[i-1];
   a4[i]=a4[i-1]+a3[i-1];
}

return  a4[n]+a3[n]+a2[n]+a1[n];

    }
int main()
{
    int n;   //年数

    while(cin>>n&&n!=0)
    {

        cout<<cow(n)<<endl;
    }
    return 0;
}