递归递推 A

A 母牛的故事

有一头母牛，第二年年初开始每年生一头小母牛，每头小母牛第四年成熟为母牛，问第n年有多少母牛。

用a【n】表示母牛数量，b【n】表示小母牛数量。易知初始状态下， a[1]=1;a[2]=1;a[3]=1; b[1]=0;b[2]=1;b[3]=1;

递推关系式为 a[i]=a[i-1]+b[i-3]; b[i]=a[i];

代码如下：

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,i,j;

	while(cin>>n&&n!=0)
	{
		if(n==1)	cout<<"1"<<endl;
		else if(n==2)	cout<<"2"<<endl;
		else
		{
			int a[55],b[55];
			a[1]=1;a[2]=1;a[3]=1;
			b[1]=0;b[2]=1;b[3]=1;
			for(i=4;i<=n;i++)
			{
				a[i]=a[i-1]+b[i-3];
				b[i]=a[i];
			}
			cout<<a[n]+b[n]+b[n-1]+b[n-2]<<endl;

		}


	}
	
	
	
	
	return 0;
}