生成函数
在数学中,某个序列(an)的母函数(又称生成函数,英语:Generating function)是一种形式幂级数,其每一项的系数可以提供关于这个序列的信息。使用母函数解决问题的方法称为母函数方法。
母函数可分为很多种,包括普通母函数、指数母函数、L级数、贝尔级数和狄利克雷级数。对每个序列都可以写出以上每个类型的一个母函数。构造母函数的目的一般是为了解决某个特定的问题,因此选用何种母函数视乎序列本身的特性和问题的类型。母函数,又称生成函数,是ACM竞赛中经常使用的一种解题算法,常用来解决组合方面的题目。
小题
有1克、2克、3克、4克的砝码各一枚,能称出哪几种重量?每种重量各有几种可能方案?
我们用母函数来解决这个问题
1个1克砝码可以看成1+x^1,1表示不取,x^1表示取一个,以下同理
1个2克砝码可以看成1+x^2
1个3克砝码可以看成1+x^3
1个4克砝码可以看成1+x^4
那么生成函数就是
g(x)=(1+x1)(1+x2)(1+x3)(1+x4)
=1+x+x2+2x3+2x4+2x5+2x6+2x7+x8+x9+x10
这个函数中可以看出重量为3克的方案有两种,重量为7的方案有两种,重量为10的有1种。
二、
求用1分、2分、3分的邮票贴出不同数值的方案数:
大家把这种情况和第一种比较有何区别?第一种每种是一个,而这里每种是无限的。
那么生成函数就是g(x)=(1+x+x2+x3+…)(1+x2+x4+x6+…)(1+x3+x6+x9+…)
(每一个多项式的个数代表了这种东西的个数,而指数代表的是价值。)
找单词
Problem Description
假设有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
这是一个简单的生成函数的模板类型;
这也是构造生成函数,每个字母的数量就是每项有多少个数,而每一字母的价值就是指数。
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int a[30];
int a1[55],a2[55]; //a1[]用于存放多项式系数,指数为j,系数为a1[j], a2[]就是中间过渡 的,
void Mhs() //模拟多项式相乘的母函数
{
memset(a1,0,sizeof(a1));
memset(a2,0,sizeof(a2));
a1[0]=1; //x^0的系数为1
for(int i=1; i<=26; i++) //表示第几个多项式
{
for(int j=0; j<=50; j++) //只计算指数<=50的,所以控制到50,j表示是指数,一个多项式中的每一项都要和指数为0-50的数相乘,
{
for(int k=0; k<=a[i] && k*i+j<=50; k++) ;//k代表的是第i个式子中的第K个数,k*i --表示的是第i个式子中的第K个数的指数,加上j表示相乘,
{
a2[k*i+j]+=a1[j]; //指数相加得j+k*i,而加多少取决于a1[j]的系数,因为被乘多项式的系数均为1
}
}
for(int j=0; j<=50; j++)
{
a1[j]=a2[j];
a2[j]=0;
}
}
}
int main()
{
int n;
int sum;
scanf("%d",&n);
while(n--)
{
memset(a,0,sizeof(a));
sum=0;
for(int i=1; i<=26; i++)
{
scanf("%d",&a[i]);
}
Mhs();
for(int i=1; i<=50; i++) //不计算x^0的系数是因为单词至少包含一个字母
{
sum+=a1[i];
}
printf("%d\n",sum);
}
return 0;
}
上一个题目可以转化为背包问题,给出26个物品,每个物品有特定的数量和特定的价值,现有一个容量为50的背包,问有多少种装法
如果相同字母i有x个,则抽象为x+1个物品,价值分别为i0,i1,i2~in
开个二维数组,一维表示字母种类,二维表示价值,于是dp[i][j]就表示 容量为j时,使用前i个字母有多少种方法;
状态转移方程dp[i][j]+=dp[i-1][j-i*k]
#include<bits/stdc++.h>
using namespace std;
int a[100],dp[100][100];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
for(int i=1; i<=26; i++)
scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));
dp[0][0]=1;
for(int i=1; i<=26; i++)//枚举物品
{
for(int k=0;k<=a[i];k++)//枚举个数
{
for(int j=50;j>=i*k;j--)//01背包
dp[i][j]+=dp[i-1][j-i*k];
}
}
int ans=0;
for(int i=1; i<=50; i++)
ans+=dp[26][i];
printf("%d\n",ans);
}
return 0;
}