hdu 2082 找单词（母函数）

最新推荐文章于 2020-03-21 20:44:57 发布

mr_zj_

最新推荐文章于 2020-03-21 20:44:57 发布

阅读量360

点赞数

分类专栏：组合数学母函数文章标签：母函数

本文链接：https://blog.csdn.net/mr_zj_/article/details/68961454

版权

组合数学母函数专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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

解题方案：母函数

假设只有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$ ，就可以得到所有能够组成的价值（指数），以及组成这些价值的方案数（系数）。
假设只有A、B、C三个字母，且每个字母可以用无限多次，则由上面的推导，可知此时母函数为 $G(x)=(1+x+x^2+...)(1+x^2+x^4+...)(1+x^3+x^6+...)$
本题就是更加一般的情况：

定理：设从 $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;
}