奶牛問題

有一頭母牛，它每年年初生一頭小母牛。每頭小母牛從第四個年頭開始，每年年初也生一頭小母牛。假設第一年就一頭老母牛，第二年就2頭母牛，第三牛就3頭母牛，第四年就4頭母牛，第五年就6頭。第六年就9頭。請編程實現在第n年的時候，共有多少頭母牛？

#include<stdio.h>
int main() {
    int age[1000] = { 0 };  //age表示每頭牛的年齡，比如age[1],表示爲第一頭牛的年齡，爲了方便看，a[0]我就捨棄不用了
    int n;
    scanf("%d", &n);
    int count = 1;    //count表示牛總數
    age[1] = 4;
    for (int i = 2; i <= n; i++) {        //i表示年份，從第二年開始
        for (int j = 1; j <= count; j++) {    //j表示奶牛【1，count】
            age[j]++;           //每年所有牛的年齡+1
            if (age[j] >= 4)    //年齡大於等於4的可以生小牛
                count++;             //牛的總數+1
        }
    }
    printf("%d",count);
    return 0;
}