PAT练习 母牛的故事

题目描述：

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

输入描述:

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

输出描述:

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

输入例子:

2
4
5

输出例子:

2
4
6

代码：

#include<iostream>

using namespace std;

int main(){
    int a[56]={0,1,2,3,4},n;
    while(~scanf("%d",&n)){
        for(int i=5;i<=n;i++){
            a[i] = a[i-1] + a[i-3];
        }
        cout<<a[n]<<endl;
    }
    return 0;
}