HDU 1085 (母函数)

/*


先初始化,再一个一个来;

注意i,j,k范围的变化;

参考大牛资料

母函数(Generating function)详解 — TankyWoo




*/


#include <iostream>

using namespace std;

int a[10005],b[10005];
int main()
{
	int num1,num2,num3;
	while(cin>>num1>>num2>>num3)
	{
		if(num1==0&&num2==0&&num3==0)
		{
			break;
		}
		int sum = num1*1 + num2*2 + num3*5;
		for(int i = 0; i <= sum; i++)     //表达式的初始化
		{
			a[i] = b[i] = 0;
		}
		
		for(int i = 0; i <= num1; i++)
		{
			a[i] = 1;
		}
		for(int i = 0; i <= num1; i++)
		for(int j = 0; j <= num2*2; j+=2)
		{
			b[j+i]+= a[i];
		}
		
		for(int i = 0; i <= num1*1+num2*2; i++)
		{
			a[i] = b[i];
			b[i] = 0;
		}
		for(int i = 0; i <= num1*1+num2*2; i++)
		for(int j = 0; j <= num3*5; j+= 5)
		{
			b[j+i] += a[i];
		}
		for(int i = 0; i <= sum; i++)
		{
			a[i] = b[i];
		}
		int p = 0;
		for(int i = 0; i <= sum; i++)
		{
			if(!a[i])
			{
				cout<<i<<endl;
				p = 1;
				break;
			}
		}
		if(!p)
		{
			cout<<sum+1<<endl;
		}
	}
}


你可能感兴趣的:(HDU 1085 (母函数))