ECNU #1030——母牛生小牛

ECNU #1030——母牛生小牛

 ECNU #1030

升级题目将母牛每年生小牛的时间设为输入值，如何做？

#include <string>
#include <iostream>
#define MAX 100
using namespace std;int main(){int m;cout << "输入第几年开始每年产一头小牛： ";cin >> m;int count[MAX] = {0};int last = 0;int amount = 0; // 总数int n; // 年数while(cin >> n && n>=0 && n<=50){amount = 0;for(int i=0; i<m; i++)count[i] = 0;count[1] = 1;last = 0;if(n == 0)break;else if(n > 1)for(int i=2; i<=n; i++){for(int j=m-1; j>0; j--){if(j == m-1){last = count[j];count[0] += last;}count[j] = count[j-1];}/*for(int k=0; k<=m; k++)cout << count[k] << " ";cout <<endl;*/}for(int i=0; i<m; i++)amount += count[i];cout << amount << endl;}return 0;
}