母函数及其应用

转自http://blog.csdn.net/ydykl/article/details/6655142

网上找的一篇讲组合数学的文章，讲的很不错

 

 先来说一说母函数，今天是第一次学。杭电关于母函数的PPT感觉不错，挺适合入门看看的。

什么是母函数？对于序列a0，a1，a2，…构造一函数：G(x)=a0+a1*x+a2*x^2+...G(x)就是序列a0,a1...的母函数。如若已知序列a0，a1，a2，…则对应的母函数G(x)便可根据定义给出。 反之，如若已经求得序列的母函数G(x)，则该序列也随之确定。 序列a0，a1，a2，…可记为{an} 

如何用母函数来解决诸如整数拆分、邮票组合、砝码称重一类的问题？这一类问题大致是需要找到整数拆分的方案数，邮票可以组合出多少面额，某一种重量可以由多少种砝码组合来完成称重……当然利用类似动态规划的思想往往也可以解决这类问题，下面讨论怎么使用母函数来完成。

下面这个例子来自HDU集训的PPT

“例1：若有1克、2克、3克、4克的砝码各一 枚，能称出哪几种重量？各有几种可能方案？ 
如何解决这个问题呢？考虑构造母函数。如果用x的指数表示称出的重量，则：
  1个1克的砝码可以用函数1+x表示，1个2克的砝码可以用函数1+x2表示，
  1个3克的砝码可以用函数1+x3表示，1个4克的砝码可以用函数1+x4表示，
几种砝码的组合可以称重的情况，可以用以上几个函数的乘积表示：
(1+x)(1+x^2)(1+x^3)(1+x^4)
=(1+x+x^2+x^3)(1+x3+x^4+x^7)
=1+x+x^2+2x^3+2x^4+2x^5+2x^6+2x^7+x^8+x^9+x^10  
从上面的函数知道：可称出从1克到10克，系数便是方案数。
”

理解了这个例子之后整数拆分、邮票组合、砝码称重一类的问题就都一并解决了。

下面来是我对这种母函数构造方式的理解。

对于上面的例1，如果取消砝码个数的限制，则母函数变为G(x)=(1+x+x^2+,,,,,)(1+x^2+x^4+...)(1+x^3+x^6+...)(1+x^4+x^8+..)

若要称量重量M，那么这个质量M就对应母函数展开后的x^M项。而x^M项的指数M按照数学上的展开来理解是什么来的呢？

假设M=14，我们用1个1g砝码，1个2g砝码，1个3g砝码和2个4g砝码。则可以看做是从G(x)=(1+x+x^2+,,,,,)(1+x^2+x^4+...)(1+x^3+x^6+...)(1+x^4+x^8+..)的每一个括号(每个括号分别对应1g, 2g, 3g, 4g砝码单独可以组合出哪些质量)里取出了x, x^2, x^3, x^8。然后意会一下为什么x^M项对应的系数就是称量质量M的方案数了。如果理解的话这一类问题就都可以做了。

实现起来，展开G(x)函数的时候就是按照平时自己手工怎么展开来做的。

两个母函数练手题，性质都是一样的

HDU1085（母函数） Holding Bin-Laden Captive!

http://acm.hdu.edu.cn/showproblem.php?pid=1085

硬币面额1,2,5且有数量限制num1,num2,num3，问最小不能组合的数量是多少。

G(x)=(1+x+...+x^num1)(1+x^2+...+x^2num2)(1+x^5+,,,+x^5num3)，展开，系数不为0的数都是可以由硬币组合出来的。

#include <cstdio>
#include <cstring>
using namespace std;

const int maxexp=1*1000+2*1000+5*1000+10;//可能的最大系数

int main()
{
    int n1, n2, n3;
    bool f;
    int c1[maxexp+1], c2[maxexp+1], c3[maxexp+1]; 
    //c[i]表示x的i次方的系数
    while (scanf("%d%d%d", &n1, &n2, &n3)==3 && n1+n2+n3>0)
    {
        memset(c1, 0, sizeof(c1));
        memset(c2, 0, sizeof(c2));
        memset(c3, 0, sizeof(c3));
        for (int i=0; i<=n1; i++)  //1+x+x^2+...+x^n1
            c1[i]=1;        //开始时系数全为一
        for (int j=0; j<=n1; j++)
            for (int k=0; k<=2*n2; k+=2)  //1+x^2+...+x^(2*n2)
                    c2[j+k]+=c1[j];
        for (int j=0; j<=n1+2*n2; j++)    //此时x的幂最大是n1+2*n2
            for (int k=0; k<=5*n3; k+=5)  //1+x^2+...+x^(5*n2)
                    c3[j+k]+=c2[j];
        f=false;
        for (int j=0; j<=n1+2*n2+5*n3; j++)//n1+2*n2+3*n3是此时x的最大幂
            if (c3[j]==0)
            {
                printf("%d\n", j);
                f=true;
                break;
            }
        if (!f) printf("%d\n", n1+2*n2+5*n3+1);
    }
    return 0;
}

HDU1028（母函数） Ignatius and the Princess III

http://acm.hdu.edu.cn/showproblem.php?pid=1028

整数拆分。由于拆分结果只考虑有几个1几个2几个3...不考虑顺序什么的，那么问题就和之前的砝码邮票什么的一样了。

G(x)=(1+x+...)(1+x^2+...)(1+x^3....)...(1+x^n)

n最大120，可以预处理一下保存结果，也可以一边输入一边重新算，反正n很小……

#include <cstdio>
#include <cstring>
using namespace std;

const int maxexp=120;

int main()
{
    int n, now, pre;
    int f[2][maxexp+1];                         //"滚筒"使用 
    
    while (scanf("%d", &n)==1)
    {
        memset(f, 0, sizeof(f));
        for (int i=0; i<=n; i++)                //1+x+x^2+...+x^n
            f[1%2][i]=1;
        for (int i=2; i<=n; i++)
        {
            now=i%2; pre=(i-1)%2;
            for (int j=0; j<=n; j++)
                f[now][j]=0;
            for (int j=0; j<=n; j++)            //枚举之前结果
                for (int k=0; k+j<=n; k+=i)     //枚举当前多项式 1+x^i+x^2i+... 
                {
                    f[now][j+k]+=f[pre][j];             //中间结果保存在c2 
                }
        }
        printf("%d\n", f[n%2][n]);
    }
    return 0;
}