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

 1 #include <iostream>
 2 
 3 int main(int argc, const char * argv[]) {
 4 
 5     int N = 0;
 6 
 7     std::cout<<"请输入年份\n";
 8     std::cin>>N;
 9     int *iA;
10     
11     if((iA = (int *)malloc(N*sizeof(int)))==NULL)
12     {
13         return 0;
14     }
15 
16     int theResult = 0;
17     
18     
19     for(int i=0;i<N;++i)
20     {
21         iA[i] = 0;
22     }
23     
24     for(int i=0;i<N;++i)
25     {
26         theResult+=1;
27         for(int j=0;j<=i;++j)
28         {
29             iA[j] += 1;
30             if(iA[j]>=4)
31                 theResult += 1;
32         }
33     }
34     
35     std::cout << theResult << "\n";
36     free(iA);
37     return 0;
38 }