1004: [递归]母牛的故事

题目
有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。请编程实现在第n年的时候，共有多少头母牛？
输入
输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。
n=0表示输入数据的结束，不做处理。
输出
对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。
输入
2
4
5
0
输出
2
4
6

#include<iostream>
using namespace std;
typedef long long LL;
LL a[1000005];
int main()
{
    int n,i,sum;
    while(cin>>n&&n>0&&n<55){
        if(n==0) break;
        a[1]=1;
        a[2]=2;
        a[3]=3;
        a[4]=4;
        if(n==1) cout<<1<<endl;
        if(n==2) cout<<2<<endl;
        if(n==3) cout<<3<<endl;
        if(n==4) cout<<4<<endl;
        if(n>=5)
        {
            sum=0;
            for(i=5;i<=n;i++){
                a[i]=a[i-1]+a[i-3];
                if(i==n) sum+=a[i];

            }
             cout<<sum<<endl;
    }
    }


    return 0;
}

思路：
年数 母牛数
1 1
2 2
3 3
4 4
5 6
6 9
索引是从1开始，第5年母牛数=第2年+第4年
第6年母牛数=第3年+第5年
所以是代码中的a[i]=a[i-1]+a[i-3];语句表示