HDU 2082 找单词 (母函数入门)

【题目链接】click here~~

【题目大意】

假设有x1个字母A， x2个字母B,..... x26个字母Z，同时假设字母A的价值为1，字母B的价值为2,..... 字母Z的价值为26。那么，对于给定的字母，可以找到多少价值<=50的单词呢？单词的价值就是组成一个单词的所有字母的价值之和，比如，单词ACM的价值是1+3+14=18，单词HDU的价值是8+4+21=33。(组成的单词与排列顺序无关，比如ACM与CMA认为是同一个单词）。

Input

输入首先是一个整数N，代表测试实例的个数。

然后包括N行数据，每行包括26个<=20的整数x1,x2,.....x26.

Output

对于每个测试实例，请输出能找到的总价值<=50的单词数,每个实例的输出占一行。

Sample Input

2
1 1 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
9 2 6 2 10 2 2 5 6 1 0 2 7 0 2 2 7 5 10 6 10 2 10 6 1 9

Sample Output

7
379297

【解题思路】母函数第一题！母函数入门详解：点击这里

代码：

<span style="font-size:14px;">/*
Author :HRW
母函数
*/
#include <bits/stdc++.h>
using namespace std;
int C1[55],C2[55],NUM[27];
int main()
{
    //freopen("1.txt","r",stdin);
    int t;
    cin>>t;
    while(t--)
    {
        memset(C1,0,sizeof(C1));
        memset(C2,0,sizeof(C2));
        for(int i=1; i<=26; i++) cin>>NUM[i];
        C1[0]=1;   //相当于用X^0去乘以后面的多项式
        for(int i=1; i<=26; i++){  //要乘以26个多项式
            for(int j=0; j<=50; j++){//c1的各项的指数
                for(int k=0; k<=NUM[i]&&j+k*i<=50; k++) //k*i表示被乘多项式各项的指数,(X^0*i + X^1*i + X^2*i + ……)
                    C2[j+k*i]+=C1[j]; //指数相加得j+k*i,加多少只取决于c1[j]的系数，因为被乘多项式的各项系数均为1
            }
            memcpy(C1,C2,sizeof(C2));
            memset(C2,0,sizeof(C2));
        }
        int res=0;
        for(int i=1; i<=50; i++)  res+=C1[i];
        printf("%d\n",res);
    }
    return 0;
}
</span>