首页 > 代码库 > hdu 2082 找单词

hdu 2082 找单词

母函数问题,今天闲来无事学习了下。

在这里多项式的写法和砝码样例的写法不一样

假设有2个a,则要写成1+x^1+x^2;

假设有2个c,则要写成1+x^3+x^6+x^9;

理解了母函数,那么就没有什么困难了

#include<iostream>
#include<cstring>
using namespace std;
int z[30],a[55],b[55];
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		memset(a,0,sizeof(a));
		memset(b,0,sizeof(b));
		a[0]=1;
		for(int i=1;i<=26;i++) cin>>z[i]; 
		for(int i=1;i<=26;i++)
		{
			if(z[i])
			{
				for(int j=0;j<=50;j++)
				{
					for(int k=0;k<=z[i]&&k*i+j<=50;k++)
					{
						b[k*i+j]+=a[j];
					}
				}
				for(int j=0;j<=50;j++) a[j]=b[j];
				memset(b,0,sizeof(b));
			}
		}
		int sum=0;
		for(int i=1;i<=50;i++) sum+=a[i];
		cout<<sum<<endl;
	}
	return 0;
} 



hdu 2082 找单词