指数型生成函数学习笔记
指数型生成函数
用于解决排列计数问题,
其生成函数形式如下:
\[ g^{(e)}(x)=1+x+\frac{x^2}{2!}+\frac{x^3}{3!}+\frac{x^4}{4!}+... \]
因为全排列出来有重复元素,对于重复了k次的a元素,在n个数中的排列方法为\(\frac{n!}{k!}\),然后有
\[ \frac{x^{m_{1}+m_{2}+\ldots+m_{k}}}{m_{1} ! m_{2} ! \ldots m_{k} !}=\frac{x^{n}}{m_{1} ! m_{2} ! \ldots m_{k} !}=\frac{n ! x^{n}}{m_{1} ! m_{2} ! \ldots m_{k} ! n !}=\frac{n !}{m_{1} ! m_{2} ! \ldots m_{k} !} \frac{x^{n}}{n !} \]
所以每一项为
\[ \frac{x^{m_{1}+m_{2}+\ldots+m_{k}}}{m_{1} ! m_{2} ! \ldots m_{k} !} \]
系数就是下面那个,就是对应的要除以的数
有一些套路公式:
\[ g^{(e)}(x)=1+x+\frac{x^2}{2!}+\frac{x^3}{3!}+\frac{x^4}{4!}+...=e^x \]
\[ g^{(e)}(x)=1-x+\frac{x^2}{2!}-\frac{x^3}{3!}+\frac{x^4}{4!}+...e^{-x} \]
\[ g^{(e)}(x)=1+\frac{x^2}{2!}+\frac{x^4}{4!}+\frac{x^6}{6!}+...=\frac{e^x+e^{-x}} 2 \]
\[ g^{(e)}(x)=x+\frac{x^3}{3!}+\frac{x^5}{5!}+\frac{x^7}{7!}+\frac{x^9}{9!}+...=\frac{e^x-e^{-x}} 2 \]
\[ x-\frac{x^2}{2}+\frac{x^3}{3}-\frac{x^4}{4}+...=\ln(x+1) \]
\[ g^{(e)}(x)=x-\frac{x^3}{3!}+\frac{x^5}{5!}-\frac{x^7}{7!}=\sin x \]
\[ g^{(e)}(x)=1-\frac{x^2}{2!}+\frac{x^4}{4!}-\frac{x^6}{6!}=\cos x \]
\[ 1+\frac{a}{1!}x+\frac{a(a-1)}{2!}x^2+\frac{a(a-1)(a-2)}{3!}x^3+...=(1+x)^a \]
练习
POJ-3734
选4种颜色,总数量为n,要求有红色和绿色必须是偶数个
由于是排列问题,需要用到指数型生成函数
对于相同颜色的,我们可以放到一起讨论
列出来的式子即为:
\[ g^{(e)}(x)=(1+\frac{x^2}{2!}+\frac{x^4}{4!}+...)^2*(1+x+\frac{x^2}{2!}+\frac{x^3}{3!}+...)^2=(\frac{e^x+e^{-x}} 2)*(e^x)^2 \]
\[ =\frac {e^{4x}+2e^{2x}+1} 4 \]
由于
\[ e^{ax}=\sum_{n=0}^{\infty}\frac{(ax)^n}{n!}=\sum_{n=0}^{\infty}a^n\frac{x^n}{n!} \]
即为数列\(1,a^1,a^2,a^3,...\)的指数型生成函数,
所以
\[ g^{(e)}(x)=\frac 1 4+\sum_{n=0}^{\infty}\frac{4^n+2^{n+1}}{4}\frac{x^n}{n!} \]
第n项的系数为\(\frac{4^n+2^{n+1}}{4}=4^{n-1}+2^{n-1}\)
代码
#include
#include
#include
#include
using namespace std;
int read(){
int x=0,pos=1;char ch=getchar();
for(;!isdigit(ch);ch=getchar()) if(ch=='-') pos=0;
for(;isdigit(ch);ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
return pos?x:-x;
}
int T;
const int mod=10007;
int ksm(int a,int b){
int res=1;
while(b){
if(b&1) res=(res*a)%mod;
a=(a*a)%mod;
b>>=1;
}
return res;
}
int main(){
T=read();
while(T--){
int n=read();
printf("%d\n",(ksm(4,n-1)+ksm(2,n-1))%mod);
}
return 0;
}
在vjudge的leaderboard上rank21
HDU1521
板子题,m^2暴力展开就行了
注意指数型生成函数求出的是下面要除的数,需要乘上\(m!\)
#include
#include
#include
#include
using namespace std;
int read() {
int x=0,pos=1;
char ch=getchar();
for(; !isdigit(ch); ch=getchar()) if(ch=='-') pos=0;
for(; isdigit(ch); ch=getchar()) x=(x<<1)+(x<<3)+ch-'0';
return pos?x:-x;
}
const int N = 20;
double f[N];
double a[N],now[N],nex[N];
int main() {
f[0]=1;
for(int i=1; i<=10; i++) {
f[i]=f[i-1]*i;
}
int n,m;
while(scanf("%d%d",&n,&m)!=EOF) {
memset(a,0,sizeof(a));
a[0]=1;
for(int i=1; i<=n; i++) {
int x=read();
x=min(x,m);
memset(nex,0,sizeof(nex));
now[0]=1;
for(int j=1; j<=x; j++) {
now[j]=1.0/f[j];
}
for(int j=0; j<=m; j++) {
for(int k=0; k<=x; k++) {
if(j+k<=m) {
nex[j+k]+=(a[j]*now[k]);
}
}
}
for(int j=0; j<=m; j++) {
a[j]=nex[j];
}
}
printf("%.0f\n",a[m]*f[m]);
}
return 0;
}