奶牛问题

有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。假设第一年就一头老母牛，第二年就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;
}