1004：蓝桥杯 母牛的故事

题目描述

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

输入

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

输出

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

样例输入

2
4
5
0

样例输出

2
4
6

代码：

#include <iostream>

using namespace std;
int main() {
    int n,m;

    while (1) {
        cin >> n;
        if (n == 0)break;
        if (n <= 4) {
            cout << n<<endl;
        }
        else {
            int a[60] = { 0 };
            a[1] = 1;
            a[2] = 2;
            a[3] = 3;
            a[4] = 4;
            for (int i = 5; i <= n; i++) {
                a[i] = a[i - 1] + a[i - 3];//在四年后每一年的母牛数量是上一年的牛数加上三年前的（都已经满四岁）
                m = a[i];
            }
            cout << m<<endl;
        }
    }
    return 0;
}