升级题目将母牛每年生小牛的时间设为输入值,如何做?
#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;
}