hdu 2082 找单词(母函数)

代码:

#include<cstdio>
using namespace std;

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int a[27];
        for(int i=1;i<=26;i++)
            scanf("%d",&a[i]);
        long long c1[55],c2[55];
        for(int i=0;i<=50;i++)
        {
            c1[i]=c2[i]=0;
        }
        c1[0]=1;
        for(int i=1;i<=26;i++)
        {
            if(a[i]==0)
                continue;
            for(int j=0;j<=50;j++)
            {
                for(int k=0;k<=i*a[i]&&k+j<=50;k+=i)
                {
                    c2[j+k]+=c1[j];
                }
            }
            for(int j=0;j<=50;j++)
            {
                c1[j]=c2[j];
                c2[j]=0;
            }
        }
        long long ans=0;
        for(int i=1;i<=50;i++)
        {
            ans+=c1[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}


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