生成函数Euler变换学习笔记(无标号有根树计数)
眾所周知,對(duì)于有標(biāo)號(hào)計(jì)數(shù)的指數(shù)型生成函數(shù) f(x)f(x)f(x),將其任意地進(jìn)行無順序的組合,得到的生成函數(shù)是exp(f(x))exp(f(x))exp(f(x))。
而對(duì)于無標(biāo)號(hào)計(jì)數(shù)的這樣的組合,我們就需要引入一個(gè)叫 Eular\text{Eular}Eular 變換的東西 E(f(x))\mathcal{E}(f(x))E(f(x))
(這個(gè)符號(hào)打法:\mathcal(E))
為了方便,本文中記fif_ifi?表示[xi]f(x)[x^i]f(x)[xi]f(x)。兩種表述可能混用。
我們先引入一個(gè)實(shí)際問題:nnn個(gè)點(diǎn)無標(biāo)號(hào)有根樹的計(jì)數(shù)。兩棵樹被認(rèn)為相同,當(dāng)且僅當(dāng)存在一種重標(biāo)號(hào)方案,使得它們的根標(biāo)號(hào)相同且邊集相等。
對(duì)于一棵樹,我們?nèi)サ羲母?#xff0c;然后就分成若干棵獨(dú)立的子樹。所以對(duì)無標(biāo)號(hào)有根樹的OGF\text{OGF}OGF F(x)F(x)F(x),有
F(x)=x?E(F(x))F(x)=x\cdot\mathcal E(F(x))F(x)=x?E(F(x))
先思考這個(gè)問題的主要矛盾在什么地方:對(duì)于分出來的若干子樹,如果有一些子樹完全一致,那么它們只會(huì)貢獻(xiàn)一次方案。但如果只是大小相同而本質(zhì)不同,仍然算作不同的方案。
換句話說,最終的方案只和每種本質(zhì)不同的子樹使用的個(gè)數(shù)有關(guān)。考慮背包。
對(duì)每種本質(zhì)不同的大小為kkk的樹構(gòu)造完全背包的生成函數(shù),即
11?xk\frac{1}{1-x^k}1?xk1?
那么我們可以得到一個(gè)簡單粗暴的定義式:
E(F(x))=∏i=1∞1(1?xi)Fi\mathcal E(F(x))=\prod_{i=1}^{\infin}\frac 1{(1-x^i)^{F_i}}E(F(x))=i=1∏∞?(1?xi)Fi?1?
FiF_iFi?是因?yàn)橛?span id="ozvdkddzhkzd" class="katex--inline">FiF_iFi?種本質(zhì)不同大小為iii的物品
然后開始愉快地推式子
G(x)=E(F(x))=∏i=1∞1(1?xi)FiG(x)=\mathcal E(F(x))=\prod_{i=1}^{\infin}\frac 1{(1-x^i)^{F_i}}G(x)=E(F(x))=i=1∏∞?(1?xi)Fi?1?
既然有連乘,先求個(gè)ln?\lnln
ln?(G(x))=?∑i=1∞Filn?(1?xi)\ln (G(x))=-\sum_{i=1}^{\infin}F_i\ln(1-x^i)ln(G(x))=?i=1∑∞?Fi?ln(1?xi)
ln?\lnln還是看不順眼,求個(gè)導(dǎo)
G′(x)G(x)=∑i=1∞Fiixi?11?xi\frac{G'(x)}{G(x)}=\sum_{i=1}^{\infin}F_i\frac{ix^{i-1}}{1-x^i}G(x)G′(x)?=i=1∑∞?Fi?1?xiixi?1?
又看見了喜聞樂見的完全背包,直接展開
G′(x)G(x)=∑i=1∞Fi?ixi?1∑j=0∞xij\frac{G'(x)}{G(x)}=\sum_{i=1}^{\infin}F_i\cdot ix^{i-1}\sum_{j=0}^{\infin}x^{ij}G(x)G′(x)?=i=1∑∞?Fi??ixi?1j=0∑∞?xij
G′(x)G(x)=∑i=1∞Fi?i∑j=0∞xi(j+1)?1\frac{G'(x)}{G(x)}=\sum_{i=1}^{\infin}F_i\cdot i\sum_{j=0}^{\infin}x^{i(j+1)-1}G(x)G′(x)?=i=1∑∞?Fi??ij=0∑∞?xi(j+1)?1
G′(x)G(x)=∑i=1∞Fi?i∑j=1∞xij?1\frac{G'(x)}{G(x)}=\sum_{i=1}^{\infin}F_i\cdot i\sum_{j=1}^{\infin}x^{ij-1}G(x)G′(x)?=i=1∑∞?Fi??ij=1∑∞?xij?1
再積分回來
ln?(G(x))=∑i=1∞Fi?i∑j=1∞xijij\ln(G(x))=\sum_{i=1}^{\infin}F_i\cdot i\sum_{j=1}^{\infin}\frac{x^{ij}}{ij}ln(G(x))=i=1∑∞?Fi??ij=1∑∞?ijxij?
整理一下
ln?(G(x))=∑i=1∞Fi∑j=1∞xijj\ln(G(x))=\sum_{i=1}^{\infin}F_i\sum_{j=1}^{\infin}\frac{x^{ij}}jln(G(x))=i=1∑∞?Fi?j=1∑∞?jxij?
ln?(G(x))=∑j=1∞1j∑i=1∞Fixij\ln(G(x))=\sum_{j=1}^{\infin}\frac 1j\sum_{i=1}^{\infin}F_ix^{ij}ln(G(x))=j=1∑∞?j1?i=1∑∞?Fi?xij
ln?(G(x))=∑j=1∞F(xj)j\ln(G(x))=\sum_{j=1}^{\infin}\frac{F(x^j)}jln(G(x))=j=1∑∞?jF(xj)?
ln?(G(x))=∑i=1∞F(xi)i\ln(G(x))=\sum_{i=1}^{\infin}\frac{F(x^i)}iln(G(x))=i=1∑∞?iF(xi)?
得到 Eular\text{Eular}Eular 變換的計(jì)算式(霧)
E(F(x))=exp?(∑i=1∞F(xi)i)\mathcal E(F(x))=\exp(\sum_{i=1}^{\infin}\frac{F(x^i)}{i})E(F(x))=exp(i=1∑∞?iF(xi)?)
然后回到原問題
F(x)=x?E(F(x))F(x)=x\cdot\mathcal E(F(x))F(x)=x?E(F(x))
F(x)=x?exp?(∑i=1∞F(xi)i)F(x)=x\cdot \exp(\sum_{i=1}^{\infin}\frac{F(x^i)}{i})F(x)=x?exp(i=1∑∞?iF(xi)?)
兩邊求導(dǎo)(如果高數(shù)不好建議自己多寫幾步)
F′(x)=exp?(∑i=1∞F(xi)i)+x?exp?(∑i=1∞F(xi)i)?∑i=1∞ixi?1?F′(xi)iF'(x)=\exp(\sum_{i=1}^{\infin}\frac{F(x^i)}{i})+x\cdot \exp(\sum_{i=1}^{\infin}\frac{F(x^i)}{i})\cdot\sum_{i=1}^{\infin}\frac{ix^{i-1}\cdot F'(x^i)}iF′(x)=exp(i=1∑∞?iF(xi)?)+x?exp(i=1∑∞?iF(xi)?)?i=1∑∞?iixi?1?F′(xi)?
F′(x)=exp?(∑i=1∞F(xi)i)+F(x)∑i=1∞xi?1?F′(xi)F'(x)=\exp(\sum_{i=1}^{\infin}\frac{F(x^i)}{i})+F(x)\sum_{i=1}^{\infin}x^{i-1}\cdot F'(x^i)F′(x)=exp(i=1∑∞?iF(xi)?)+F(x)i=1∑∞?xi?1?F′(xi)
兩邊同時(shí)乘上xxx
x?F′(x)=F(x)+F(x)∑i=1∞xi?F′(xi)x\cdot F'(x)=F(x)+F(x)\sum_{i=1}^{\infin}x^i\cdot F'(x^i)x?F′(x)=F(x)+F(x)i=1∑∞?xi?F′(xi)
#undef G
記
G(x)=∑i=1∞xi?F′(xi)G(x)=\sum_{i=1}^{\infin}x^i\cdot F'(x^i)G(x)=i=1∑∞?xi?F′(xi)
x?F′(x)=F(x)+F(x)G(x)x\cdot F'(x)=F(x)+F(x)G(x)x?F′(x)=F(x)+F(x)G(x)
考慮求這東西的第xnx^nxn項(xiàng)
nFn=Fn+∑k=1n?1FkGn?knF_n=F_n+\sum_{k=1}^{n-1}F_kG_{n-k}nFn?=Fn?+k=1∑n?1?Fk?Gn?k?
Fn=1n?1∑k=1n?1FkGn?kF_{n}=\frac 1{n-1}\sum_{k=1}^{n-1}F_kG_{n-k}Fn?=n?11?k=1∑n?1?Fk?Gn?k?
如果我們知道GGG,就可以直接分治 NTT 算出FFF
現(xiàn)在考慮這個(gè)GGG
G(x)=∑i=1∞xi?F′(xi)G(x)=\sum_{i=1}^{\infin}x^i\cdot F'(x^i)G(x)=i=1∑∞?xi?F′(xi)
Gn=∑i=1∞[xn?i]F′(xi)G_n=\sum_{i=1}^{\infin}[x^{n-i}]F'(x^i)Gn?=i=1∑∞?[xn?i]F′(xi)
經(jīng)過大膽的觀察和仔細(xì)的猜想
Gn=∑d∣n(n?dd+1)Fn?dd+1G_n=\sum_{d\mid n}(\frac {n-d}d+1)F_{\frac {n-d}d+1}Gn?=d∣n∑?(dn?d?+1)Fdn?d?+1?
Gn=∑d∣n(nd)FndG_n=\sum_{d\mid n}(\frac nd)F_{\frac nd}Gn?=d∣n∑?(dn?)Fdn??
Gn=∑d∣ndFdG_n=\sum_{d\mid n}d F_dGn?=d∣n∑?dFd?
然后每求出一個(gè)FFF,就可以調(diào)和級(jí)數(shù)復(fù)雜度求出之前的GGG,就可以分治 NTT 了。
但是分治時(shí)如果L=1L=1L=1,GGG可能不夠用。解決辦法是先不計(jì)算不夠的地方的貢獻(xiàn),在后面再把它加上。
具體的講,如果我們要計(jì)算F1~midF_{1\sim mid}F1~mid?對(duì)Fmid+1~RF_{mid+1\sim R}Fmid+1~R?的貢獻(xiàn),我們只計(jì)算iii不超過midmidmid的GiG_iGi?的貢獻(xiàn),這樣少算了Gmid+1,R?1G_{mid+1,R-1}Gmid+1,R?1?的貢獻(xiàn)。而這部分貢獻(xiàn)可以通過RRR之前交換FFF和GGG做一次類似的卷積計(jì)算出來。詳見代碼。
復(fù)雜度O(nlog?2n)O(n\log ^2n)O(nlog2n)
#include <iostream> #include <cstdio> #include <cstring> #include <cctype> #define MAXN 524288 using namespace std; const int MOD=998244353; typedef long long ll; inline int add(const int& x,const int& y){return x+y>=MOD? x+y-MOD:x+y;} inline int dec(const int& x,const int& y){return x<y? x-y+MOD:x-y;} inline int qpow(int a,int p) {int ans=1;while (p){if (p&1) ans=(ll)ans*a%MOD;a=(ll)a*a%MOD;p>>=1;}return ans; } #define inv(x) qpow(x,MOD-2) int rt[2][24],r[MAXN],l,lim; inline void init(){lim=1<<l;for (int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));} inline void NTT(int* a,int type) {for (int i=0;i<lim;i++) if (i<r[i]) swap(a[i],a[r[i]]);for (int L=0;L<l;L++){int mid=1<<L,len=mid<<1;ll Wn=rt[type][L+1];for (int s=0;s<lim;s+=len)for (int k=0,w=1;k<mid;k++,w=w*Wn%MOD){int x=a[s+k],y=(ll)w*a[s+mid+k]%MOD;a[s+k]=add(x,y),a[s+mid+k]=dec(x,y);}}if (type){int t=inv(lim);for (int i=0;i<lim;i++) a[i]=(ll)a[i]*t%MOD;} } int f[MAXN],g[MAXN],n; int a[MAXN],b[MAXN]; void solve(int L,int R) {if (L==R){if (L>1) f[L]=(ll)f[L]*inv(L-1)%MOD;int t=(ll)L*f[L]%MOD;for (int i=L;i<=n;i+=L) g[i]=add(g[i],t);return;}int mid=(L+R)>>1;solve(L,mid);int len=R-L;for (l=0;(1<<l)<=(len<<1);l++);init();if (L==1){for (int i=0;i<lim;i++) a[i]=b[i]=0;for (int i=L;i<=mid;i++) a[i]=f[i],b[i]=g[i];NTT(a,0);NTT(b,0);for (int i=0;i<lim;i++) a[i]=(ll)a[i]*b[i]%MOD;NTT(a,1);for (int i=mid+1;i<=R;i++) f[i]=add(f[i],a[i]);}else{for (int i=0;i<lim;i++) a[i]=b[i]=0;for (int i=L;i<=mid;i++) a[i-L]=f[i];for (int i=0;i<=len;i++) b[i]=g[i];NTT(a,0);NTT(b,0);for (int i=0;i<lim;i++) a[i]=(ll)a[i]*b[i]%MOD;NTT(a,1);for (int i=mid+1;i<=R;i++) f[i]=add(f[i],a[i-L]);for (int i=0;i<lim;i++) a[i]=b[i]=0;for (int i=L;i<=mid;i++) a[i-L]=g[i];for (int i=0;i<=len;i++) b[i]=f[i];NTT(a,0);NTT(b,0);for (int i=0;i<lim;i++) a[i]=(ll)a[i]*b[i]%MOD;NTT(a,1);for (int i=mid+1;i<=R;i++) f[i]=add(f[i],a[i-L]);}solve(mid+1,R); } int main() {rt[0][23]=qpow(3,119),rt[1][23]=inv(rt[0][23]);for (int i=22;i>=0;i--){rt[0][i]=(ll)rt[0][i+1]*rt[0][i+1]%MOD;rt[1][i]=(ll)rt[1][i+1]*rt[1][i+1]%MOD;}scanf("%d",&n);f[1]=1;solve(1,n);int ans=f[n];printf("%d\n",ans);return 0; }總結(jié)
以上是生活随笔為你收集整理的生成函数Euler变换学习笔记(无标号有根树计数)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 山东山西的山是指哪座山 原来有两种说法
- 下一篇: 网站设计怎么自学(网站设计怎么自学的)