母函数问题,今天闲来无事学习了下。
在这里多项式的写法和砝码例子的写法不一样
如果有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;
}