hdu 2082 找单词（母函数）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2082

解题方案：母函数

1. 假设只有A、B、C三个字母，且每个字母只能用一次，可以构造出母函数$G(x)=(1+x)(1+x^2)(1+x^3)$，指数代表价值，其中$1=x^0$，代表不取，价值贡献为0。
   $(1+x)$：即$(x^0+x^1)$，代表字母A不选或者选。
   $(1+x^2)$：即$(x^0+x^2)$，代表字母B不选或者选。
   $(1+x^3)$：即$(x^0+x^3)$，代表字母C不选或者选。
   则$G(x)=(1+x)(1+x^2)(1+x^3)=1+x+x^2+2x^3+x^4+x^5+x^6$，就可以得到所有能够组成的价值（指数），以及组成这些价值的方案数（系数）。
2. 假设只有A、B、C三个字母，且每个字母可以用无限多次，则由上面的推导，可知此时母函数为$G(x)=(1+x+x^2+...)(1+x^2+x^4+...)(1+x^3+x^6+...)$

3. 本题就是更加一般的情况：

   定理：设从$n$元集合$S=\{ a_1,a_2,a_3,...,a_n \}$中取元素进行组合，若限定元素$a_i$出现的次数集合为$M_i(1\le i \le n)$，则该组合数序列的母函数为$G(x)=\prod_{i=1}^n(\sum_{m \in M_i} x^m)$

   本题只需要在上述定理中将$x$的幂$m$再乘上一个权重$V_i$，其中$V_i=i,（1 \le i \le 26）$，接下来就可以模拟计算多项式乘法了，从而得到结果。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>

using namespace std;

#define FOR(i,k,n) for(int i=k;i<n;i++)
#define FORR(i,k,n) for(int i=k;i<=n;i++)
#define scan(a) scanf("%d",&a)
#define scann(a,b) scanf("%d%d",&a,&b)
#define scannn(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define mst(a,n)  memset(a,n,sizeof(a))
#define ll long long
#define N 55
#define mod 1000000007
#define INF 0x3f3f3f3f

const double eps=1e-8;
const double pi=acos(-1.0);

int a[N];//存储多项式相乘的结果，a[i]代表指数为i的项的系数
int tmp[N];

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);

    int t;
    scan(t);
    while(t--)
    {
        mst(a,0);
        a[0]=1;//在多项式乘法链的首部添加辅助多项式(1+0*x^1+0*x^2+...+0*x^50)
        FORR(i,1,26)//乘上第i个多项式
        {
            int cnt;
            scan(cnt);
            mst(tmp,0);
            FORR(j,0,50)//模拟多项式乘法，前i-1项的51个元素和第i项cnt+1个元素相乘
            {
                FORR(k,0,cnt)
                {
                    if(j+k*i>50) break;
                    tmp[j+k*i]+=a[j];//此处数组a存着前i-1个多项式相乘的结果
                }
            }
            FORR(j,0,50) a[j]=tmp[j];//更新数组a
        }
        int ans=0;
        FORR(i,1,50) ans+=a[i];
        printf("%d\n",ans);
    }
    return 0;
}