递推-昆虫繁殖

昆虫繁殖
题目描述
科学家在热带森林中发现了一种特殊的昆虫，这种昆虫的繁殖能力很强。每对成虫每过x个月产y对卵，每对卵要过两个月长成成虫。假设每个成虫不死，第一个月只有一对成虫，且卵长成成虫后的第一个月不产卵(过X个月产卵)，问过Z个月以后，共有成虫多少对？x,y,z均为正整数
输入
输入只有一行为三个正整数：x y z 0=<X<=20,1<=Y<=20,X=<Z<=50
输出
输出只有一行，为z个月后成虫的对数（注意，表示要求第z个月结束后，即第z+1个月开始时的成虫个数）
样例输入
1 2 8
样例输出
37

一开始搞错了，错误代码如下：

#include <iostream>
using namespace std;
int x,y,z,y1,y2,a[50],ans=1;
int main()
{
	cin>>x>>y>>z;
	y1=y*ans;
	for (int i=1;i<=z;i++)
	{
		for (int j=1;j<=x;j++)	ans=ans+a[j];
		for (int j=x;j>1;j--)
		{
			a[j]=a[j-1];
		}
		a[1]=y2;
		y2=y1;
		y1=y*ans;
	}
	cout<<ans; 
	return 0;
}

后来根据数据纠正错误，AC代码如下，效率不高，但能模拟出来也不错，对吧？

#include <iostream>
using namespace std;
long long int x,y,z,y1,y2,a[50],ans=1;
int main()
{
	cin>>x>>y>>z;
	y1=y*ans;
	for (int i=1;i<=z;i++)
	{
		ans=ans+a[x];
		for (int j=x;j>1;j--)
		{
			a[j]=a[j-1];
		}
		a[1]=y2;
		y2=y1;
		y1=y*ans;
	}
	cout<<ans; 
	return 0;
}

后面进行了代码优化

优化后代码如下：

#include <iostream>
using namespace std;
long long int x,y,z,a[50];
int main()
{
	cin>>x>>y>>z;
	for (int i=0;i<=x+2;i++) a[i]=1; 
	for (int i=x+2;i<=z;i++)
	{
		a[i]=a[i-x-2]*y+a[i-1];
	}
	cout<<a[z]; 
	return 0;
}