《编程题》穷举法求N年后有多少头牛

若一头小母牛，从出生起第四个年头开始每年生一头母牛，按这个规律，第N年时有多少头母牛？

 

#include <iostream>

int main(int argc, const char * argv[]) {

    int N = 0;

    std::cout<<"请输入年份\n";
    std::cin>>N;
    int *iA;
    
    if((iA = (int *)malloc(N*sizeof(int)))==NULL)
    {
        return 0;
    }

    int theResult = 0;
    
    
    for(int i=0;i<N;++i)
    {
        iA[i] = 0;
    }
    
    for(int i=0;i<N;++i)
    {
        theResult+=1;
        for(int j=0;j<=i;++j)
        {
            iA[j] += 1;
            if(iA[j]>=4)
                theResult += 1;
        }
    }
    
    std::cout << theResult << "\n";
    free(iA);
    return 0;
}