hdu 2082 找单词 母函数

基础的母函数 已知每个字母的价值和个数，问这些字母组成的单词的价值小于50的有多少个

#include<iostream>
#include<cstring>
using namespace std;
int a[51],b[51];//记录x^0 x^1 x^2 ....的系数
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
        a[0]=1;//只有x^0的系数为1（否则计算不出结果）,其余项为0
        for(int i=1;i<=26;i++)//i代表第i个字母
        {
            int num;
            cin>>num;
            for(int j=0;j<=50;j++)// x^j
            {
                for(int k=0;k<=num&&k*i+j<=50;k++)//x^j ＊ x^(k*i)
                {
                    b[k*i+j]+=a[j];//计算新的x^（k＊i+j）的系数
                }
            }
            for(int j=0;j<=50;j++) a[j]=b[j];
            memset(b,0,sizeof(b));
        }
        int ans=0;
        for(int i=1;i<=50;i++) ans+=a[i];
        cout<<ans<<endl;
    }
}