有一头母牛,它每年年初生一头小母牛。每头小母牛从第四个年头开始,每年年初也生一头小母牛。请编程实现在第 nn 年的时候,共有多少头母牛?
输入格式
输入数据由多个测试实例组成,每个测试实例占一行,包括一个整数n(0<n<55),n的含义如题目中描述。
n=0表示输入数据的结束,不做处理。
输出格式
对于每个测试实例,输出在第 n 年的时候母牛的数量。
每个输出占一行。
样例输入
2
4
5
0
样例输出
2
4
6
第一次用的代码,提交发现超时。因为调用函数的时间复杂度太高了。
#include<iostream>
using namespace std;
int cow(int n);
int main()
{
int n;
while (cin >> n && n != 0)
cout << cow(n);
return 0;
}
int cow(int n)
{
if (n == 1)
return 1;
else if (n == 2)
return 2;
else if (n == 3)
return 3;
else if (n == 4)
return 4;
else
return cow(n - 1) + cow(n - 3);
}
将函数写为数组就好了
#include<iostream>
using namespace std;
int main()
{
int n;
int cow[60];
cow[1] = 1;
cow[2] = 2; cow[3] = 3;
for (int i = 4; i < 60; i++)
cow[i] = cow[i - 1] + cow[i - 3];
while (cin >> n && n != 0)
cout <<cow[n]<<endl;
return 0;
}