1004 母牛的故事

/*问题 1004: 母牛的故事

时间限制: 1Sec 内存限制: 128MB 提交: 17470 解决: 4679

题目描述
有一头母牛，它每年年初生一头小母牛。每头小母牛从第四个年头开始，每年年初也生一头小母牛。请编程实现在第n年的时候，共有多少头母牛？

输入
输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。
n=0表示输入数据的结束，不做处理。

输出
对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。

样例输入
2
3
4
5
样例输出：
2
3
4
6
代码一：运用递归的做法，不过超时
思路：

#include<iostream>
#define arr_size 56
using namespace std;
int cattle_num(int year_num);//函数得声明

int main(void)
{
	int n;
	int arr[arr_size]={0};
	int index=0;
	int len=0;
	do
	{
		
		cin>>arr[index];
		index++;
	}while(arr[index-1]!=0);
	len=index;//输入有效数字的个数 
	for (int i=0;i<len-1;i++) 
	{
		cout<<cattle_num(arr[i])<<endl;
	}
	return 0; 
} 
//求第n年小牛的数量 
//递归的使用 
int cattle_num(const int year_num)
{
	int year=year_num;
	if (year<=4)
	{
		return year;
	} 
	else
	{
		return cattle_num(year-1)+cattle_num(year-3);
	}
}

代码二：改用循环

#include<iostream>
#define arr_size 55
#define term_size 56
using namespace std;
int cattle_num(const int year);

int main(void)
{
	int n;
	int arr[arr_size]={1,2,3,4};
	int num_term[term_size]={0}; 
	int index=0;
	int len=0;
	do
	{
		
		cin>>num_term[index];
		index++;
	}while(num_term[index-1]!=0);
	len=index;//输入有效数字的个数 
	
	for (int i=0;i<len-1;i++) 
	{
		for (int j=4;j<=num_term[i];j++)
		{
			arr[j]=arr[j-1]+arr[j-3]; 
		}
		cout<<arr[num_term[i]-1]<<endl;
	}
	return 0; 
}