本人第一次接触母函数方法,在网上找一篇详解总是花掉不少时间。所以转载别人一篇文章,以备查看。
原文链接:http://www.cnblogs.com/hsqdboke/archive/2012/04/17/2453677.html
(原文有不少错误,本文章已改正)
在数学中,某个序列的母函数(Generating function,又称生成函数)是一种形式幂级数,其每一项的系数可以提供关于这个序列的信息。使用母函数解决问题的方法称为母函数方法。
母函数可分为很多种,包括普通母函数、指数母函数、L级数、贝尔级数和狄利克雷级数。对每个序列都可以写出以上每个类型的一个母函数。构造母函数的目的一般是为了解决某个特定的问题,因此选用何种母函数视乎序列本身的特性和问题的类型。
这里先给出两句话,不懂的可以等看完这篇文章再回过头来看:
1.“把组合问题的加法法则和幂级数的乘幂对应起来”
2.“母函数的思想很简单 — 就是把离散数列和幂级数一 一对应起来,把离散数列间的相互结合关系对应成为幂级数间的运算关系,最后由幂级数形式来确定离散数列的构造. ”
我们首先来看下这个多项式乘法:
( 1 + a 1 x ) ( 1 + a 2 x ) . . . ( 1 + a n x ) = 1 + ( a 1 + a 2 + a n ) x + ( a 1 a 2 + a 1 a 3 + . . . + a n − 1 a n ) x 2 + . . . + a 1 a 2 . . . a n x n (1+a_1x)(1+a_2x)...(1+a_nx)=1+(a_1+a_2+a_n)x+(a_1a_2+a_1a_3+...+a_{n-1}a_n)x^2+...+a_1a_2...a_nx^n (1+a1x)(1+a2x)...(1+anx)=1+(a1+a2+an)x+(a1a2+a1a3+...+an−1an)x2+...+a1a2...anxn
由此可以看出:
x的系数是a1,a2,…an 的单个组合的全体。
x^2的系数是a1,a2,…a2的两个组合的全体。
…
x^n的系数是a1,a2,….an的n个组合的全体(只有1个)。
特别地,我们有二项式公式:
( 1 + x ) n = 1 + C ( n , 1 ) x + C ( n , 2 ) x 2 + . . . C ( n , n − 1 ) x n − 1 + C ( n , n ) x n (1+x)^n=1+C(n,1)x+C(n,2)x^2+...C(n,n-1)x^{n-1}+C(n,n)x^n (1+x)n=1+C(n,1)x+C(n,2)x2+...C(n,n−1)xn−1+C(n,n)xn
对于序列a0,a1,a2,…构造一函数:
G ( x ) = a 0 + a 1 x + a 2 x 2 + a 3 x 3 + . . . a n x n + . . . G(x)=a_0+a_1x+a_2x^2+a_3x^3+...a_nx^n+... G(x)=a0+a1x+a2x2+a3x3+...anxn+...
称函数G(x)是序列a0,a1,a2,…的母函数。
下面说一个经典例子:砝码问题
有1克、2克、3克、4克的砝码各一枚,能称出哪几种重量?每种重量各有几种可能方案?
考虑用母函数来解决这个问题:
我们假设x表示砝码,x的指数表示砝码的重量,这样:
上面这四个式子懂吗?
我们拿 1 + x 2 1+x^2 1+x2来说,前面已经说过,x表示砝码,x的指数表示砝码的重量!初始状态时,这里就是一个质量为2的砝码。
那么前面的1表示什么?按照上面的理解,1其实应该写为: 1 ∗ x 0 1*x^0 1∗x0,即1代表重量为0的砝码数量为1个。
所以这里 1 + 1 ∗ x 2 = 1 ∗ x 0 + 1 ∗ x 2 1+1*x^2 = 1*x^0 + 1*x^2 1+1∗x2=1∗x0+1∗x2,即表示1个0克的砝码(相当于没有,只是形象说明)和2克的砝码,不取或取,不取则为 1 ∗ x 0 1*x^0 1∗x0,取则为 1 ∗ x 2 1*x^2 1∗x2。
接着讨论上面的 1 + x 2 1+x^2 1+x2,这里x前面的系数有什么意义?
这里的系数表示状态数(方案数)
1 + x 2 1+x^2 1+x2,也就是 1 ∗ x 0 + 1 ∗ x 2 1*x^0 + 1*x^2 1∗x0+1∗x2,也就是上面说的不取2克砝码,此时有1种状态;或者取2克砝码,此时也有1种状态。(分析!)
所以,前面说的那句话
1.“把组合问题的加法法则和幂级数的乘幂对应起来”
的意义大家可以理解了吧?
几种砝码的组合可以称重的情况,可以用以上几个函数的乘积表示:
( 1 + x ) ( 1 + x 2 ) ( 1 + x 3 ) ( 1 + x 4 ) = ( 1 + x + x 2 + x 4 ) ( 1 + x 3 + x 4 + x 7 ) = 1 + x + x 2 + 2 x 3 + 2 x 4 + 2 x 5 + 2 x 6 + 2 x 7 + x 8 + x 9 + x 10 (1+x)(1+x^2)(1+x^3)(1+x^4) =(1+x+x^2+x^4)(1+x^3+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+x)(1+x2)(1+x3)(1+x4)=(1+x+x2+x4)(1+x3+x4+x7)=1+x+x2+2x3+2x4+2x5+2x6+2x7+x8+x9+x10
从上面的函数知道:可称出从1克到10克,系数便是方案数。(!!!经典!!!)
例如右端有 2 x 5 2x^5 2x5项,即称出5克的方案有2种:5=3+2=4+1;同样,6=1+2+3=4+2;10=1+2+3+4。故称出6克的方案数有2种,称出10克的方案数有1种 。
再举一个经典例子:邮票组合
求用1分、2分、3分的邮票贴出不同数值的方案数。
这种情况和第一个例子有何区别?第一个例子每种是一个,而这里每种是无限的。
G ( x ) = ( 1 + x + x 2 + . . . ) ( 1 + x 2 + x 4 + . . . ) ( 1 + x 3 + x 6 + . . . ) G(x)=(1+x+x^2+...)(1+x^2+x^4+...)(1+x^3+x^6+...) G(x)=(1+x+x2+...)(1+x2+x4+...)(1+x3+x6+...)
在这里,x表示邮票,x的指数表示邮票总的价值,x的系数就是方案数。
以展开后的 x 4 x^4 x4为例,其系数为4,表示价值为4的邮票组合的方案数为4。即4拆分成1、2、3之和的拆分方案数为4;即:4=1+1+1+1=1+1+2=1+3=2+2.
这里引出两个概念"整数拆分"和"拆分数":
下面给出第二种无限的情况的一个模板:
#include
#include
#include
#include
using namespace std;
#define M 1000
int a[M],b[M];//a[M]中存最终项系数;b[M]中存取中间变量;
int main(){
int m,n;
int i,j,k;
cout<<"请输入最初有几项表达式相乘: "<>m;
while(1){//可以加一个跳出循环的条件
cout<<"请输入所求的指数 n 的值: ";
cin>>n;
memset(a,0,sizeof(a));//初始化
memset(b,0,sizeof(b));
a[0]=1;
for(i=1;i<=m;i++) {
//从第1项式子一直到第m项式子,邮票组合的例子中m=3
for(j=0;j<=n;j++)
//从所累乘得到的式子中指数为0遍历到指数为n
for(k=0;k+j<=n;k+=i){
//第i个多项式的指数从0开始,
//后面的每项指数依次比前面的多i,比如当i=3时,
//第3项的表达式为(1+x^3+x^6+x^9+……),
//直到所得指数的值i+j>=n退出
b[j+k]+=a[j];
//比如前面指数为1,系数为3,
//即a[1]=3 的一项和下一个表达式的指数为k=3的相乘,
//则得到的式子的系数为b[j+k]=b[4]+=a[1],
//又a[1]=3,所以指数为4的系数为b[4]=3;
}
//然后将中间变量b数组中的值依次赋给数组a,
//然后将数组b清零,继续接收乘以下一个表达式所得的值
memcpy(a,b,sizeof(b));
memset(b,0,sizeof(b));
}
printf("指数为%d的项的系数为:%d\n\n",n,a[n]);
}
return 0;
}
现在我们来看HDU上面的这道题:
http://acm.hdu.edu.cn/showproblem.php?pid=2082
就会发现这是一个典型的利用母函数方法解题的问题。贴上代码:
#include
#include
#include
using namespace std;
int a[100],b[100];
int letters[30];
int main(){
int n,i,j,k;
long long sum;
cin>>n;
while(n--){
for(i=1;i<=26;i++)
cin>>letters[i]; //注意从i=1开始存储
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
a[0] = 1;///初始化
for(i=1;i<=26;i++){//26个字母对应26个多项式
for(j=0;j<=50;j++){//每个多项式相应的指数
for(k=0;k*i+j<=50&&k<=letters[i];k++)
b[j+k*i]+=a[j];
memcpy(a,b,sizeof(b));
memset(b,0,sizeof(b));
}
sum=0;
for(i=1;i<=50;i++)
sum+=a[i];
cout<