[集训队作业2018] count(笛卡尔树,生成函数,卡特兰数)
傳送門
什么情況下兩序列同構
對于兩序列A[1,n],B[1,n]A[1,n],B[1,n]A[1,n],B[1,n],設fA(1,n)=pa,fB(1,n)=pbf_A(1,n)=p_a,f_B(1,n)=p_bfA?(1,n)=pa?,fB?(1,n)=pb?,
- 若pa=?pbp_a\not=p_bpa??=pb?,A,BA,BA,B一定不同構。
- 若pa=pbp_a=p_bpa?=pb?,
若A[1,pa?1]A[1,p_a-1]A[1,pa??1]與B[1,pb?1]B[1,p_b-1]B[1,pb??1]同構且A[pa+1,n]A[p_a+1,n]A[pa?+1,n]與B[pb+1,n]B[p_b+1,n]B[pb?+1,n]同構,則A[1,n],B[1,n]A[1,n],B[1,n]A[1,n],B[1,n]同構;
否則A,BA,BA,B不同構。
由這個過程想到什么?笛卡爾樹!
沒錯,兩序列同構當且僅當它們的笛卡爾樹形態相同。
生成函數做法
引理證明:
A={X∣序列X長度為n,序列X中的每個數都是1到m內的整數}A=\{X| 序列X長度為n,序列X中的每個數都是1到m內的整數\}A={X∣序列X長度為n,序列X中的每個數都是1到m內的整數}
B={X∣序列X長度為n,序列X中的每個數都是1到m內的整數,且序列X中的最大值為m}B=\{X| 序列X長度為n,序列X中的每個數都是1到m內的整數,且序列X中的最大值為m\}B={X∣序列X長度為n,序列X中的每個數都是1到m內的整數,且序列X中的最大值為m}
C={X∣序列X長度為n,序列X中的每個數都是1到m內的整數,且所有1到m內的整數都在序列X中出現過}C=\{X| 序列X長度為n,序列X中的每個數都是1到m內的整數,且所有1到m內的整數都在序列 X 中出現過\}C={X∣序列X長度為n,序列X中的每個數都是1到m內的整數,且所有1到m內的整數都在序列X中出現過}
可以證明n≥mn\geq mn≥m時,A=B=CA=B=CA=B=C。
證明:
首先證明A=BA =BA=B。
對于AAA中任意一個序列XXX,若XXX中的最大值為ppp,那么把XXX中的每個數都加上m?pm?pm?p,得到序列YYY ,顯然XXX和YYY同構,即X=YX=YX=Y 。YYY中最大值為mmm,即Y∈BY\in BY∈B。
對于任意 X∈AX \in AX∈A,均有X∈BX\in BX∈B,這說明A?BA\subseteq BA?B。而 B?AB\subseteq AB?A 是顯然的,因此 A=BA=BA=B。
可以利用類似的方法證明 A=CA=CA=C。
對于AAA中任意一個序列XXX,可以在同構的前提下,保持不同權值的數之間的相對大小的同時,把權值相同的數按下標順序替換成權值不同的數。這樣在n≥mn\geq mn≥m時,通過上述替換,可以使XXX中出現過的數增加至mmm種。
對于任意X∈AX\in AX∈A,均有X∈CX\in CX∈C,這說明A?CA\subseteq CA?C。C?AC\subseteq AC?A 也是顯然的,因此 A=CA =CA=C。
DP:
設f[i][j]f[i][j]f[i][j]表示n=i,m=jn=i,m=jn=i,m=j時的∣C∣|C|∣C∣,那么f[i][j]=∣C∣=∣A∣=∣B∣f[i][j]=|C|=|A|=|B|f[i][j]=∣C∣=∣A∣=∣B∣。
考慮計算∣B∣|B|∣B∣,那么最大值為jjj,枚舉下標最小的jjj的下標,得到遞推式:
f[i][j]=∑p=1if[p?1][j?1]×f[i?p][j]f[i][j]=\sum_{p=1}^{i}f[p-1][j-1]\times f[i-p][j]f[i][j]=p=1∑i?f[p?1][j?1]×f[i?p][j]
(把f[p?1][j?1]f[p-1][j-1]f[p?1][j?1]當∣A1∣|A_1|∣A1?∣,f[i?p][j]f[i-p][j]f[i?p][j]當∣A2∣|A_2|∣A2?∣)
引入生成函數:
設Fj(x)=∑i≥0f[i][j]xiF_j(x)=\sum_{i\geq 0}f[i][j]x^iFj?(x)=∑i≥0?f[i][j]xi,代入遞推式有:
Fj(x)[xi]=∑p=1iFj?1(x)[xp?1]×Fj(x)[xi?p]F_j(x)[x^i]=\sum_{p=1}^{i}F_{j-1}(x)[x^{p-1}]\times F_j(x)[x^{i-p}]Fj?(x)[xi]=p=1∑i?Fj?1?(x)[xp?1]×Fj?(x)[xi?p]
Fj(x)[xi]=∑p=0i?1Fj?1(x)[xp]×Fj(x)[xi?1?p]F_j(x)[x^i]=\sum_{p=0}^{i-1}F_{j-1}(x)[x^{p}]\times F_j(x)[x^{i-1-p}]Fj?(x)[xi]=p=0∑i?1?Fj?1?(x)[xp]×Fj?(x)[xi?1?p]
Fj(x)[xi]=(Fj?1(x)×Fj(x))[xi?1]F_j(x)[x^i]=(F_{j-1}(x)\times F_j(x))[x^{i-1}]Fj?(x)[xi]=(Fj?1?(x)×Fj?(x))[xi?1]
平凡情況有f[0][j]=1f[0][j]=1f[0][j]=1。
所以Fj(x)=Fj(x)Fj?1(x)x+1F_j(x)=F_j(x)F_{j-1}(x)x+1Fj?(x)=Fj?(x)Fj?1?(x)x+1
所以Fj(x)=11?Fj?1(x)xF_j(x)=\frac{1}{1-F_{j-1}(x)x}Fj?(x)=1?Fj?1?(x)x1?
直接做似乎很不可做,但由這個式子我們知道FjF_jFj?可以分解為AjBj\frac{A_j}{B_j}Bj?Aj??,其中Aj,BjA_j,B_jAj?,Bj?是兩個次數界為O(j)O(j)O(j)的多項式,那么考慮求Aj,BjA_j,B_jAj?,Bj?:
AjBj=11?Aj?1Bj?1x\frac{A_j}{B_j}=\frac{1}{1-\frac{A_{j-1}}{B_{j-1}}x}Bj?Aj??=1?Bj?1?Aj?1??x1?
AjBj=Bj?1Bj?1?Aj?1x\frac{A_j}{B_j}=\frac{B_{j-1}}{B_{j-1}-A_{j-1}x}Bj?Aj??=Bj?1??Aj?1?xBj?1??
于是得到Aj=Bj?1,Bj=Bj?1?Aj?1xA_j=B_{j-1},B_j=B_{j-1}-A_{j-1}xAj?=Bj?1?,Bj?=Bj?1??Aj?1?x,將轉移關系用矩陣來表示就得到:
[01?x1](Aj?1Bj?1)=(AjBj)\begin{bmatrix}0 & 1\\-x & 1 \end{bmatrix}\begin{pmatrix}A_{j-1}\\B_{j-1}\end{pmatrix}=\begin{pmatrix}A_j\\B_j\end{pmatrix}[0?x?11?](Aj?1?Bj?1??)=(Aj?Bj??)
那么可以通過矩陣快速冪來求Aj,BjA_j,B_jAj?,Bj?。但是直接在矩陣中用多項式進行計算會很麻煩,因此不妨考慮用單位根代入求點值,用IDFTIDFTIDFT插出多項式即可。復雜度O(nlogn)O(nlogn)O(nlogn)。
參考博客
卡特蘭數做法
因為兩序列同構當且僅當它們的笛卡爾樹形態相同,所以原題就是問有多少棵不同形態的節點數為nnn的笛卡爾樹滿足要求。
為了滿足題目中所給出的 “如果有多個最大值,取下標最小的” 的要求,規定笛卡爾樹上的節點xxx滿足:左兒子的值<x<x<x的值&&右兒子的值≤x\leq x≤x的值
根據上文引理的證明,
“序列中的每個數都是1到mmm內的整數,且所有111到mmm內的整數都在序列中出現過”的條件
等價于 “序列中的每個數都是1到mmm內的整數”,
所以一棵笛卡爾樹有符合條件的填數方法,當且僅當 從根節點往下走,不存在一種走法,使得往左兒子走的次數>m?1>m?1>m?1(每往左兒子走一次,相當于權值 ?1,需要有數可填)
考慮將"往左兒子走的次數≤m?1\leq m?1≤m?1"的條件進行轉換,變為更容易處理的問題:
想到多叉樹轉二叉樹的方法:節點xxx的左兒子為原樹上xxx的兒子,右兒子為原樹上xxx的兄弟
考慮把笛卡爾樹這棵二叉樹轉回多叉樹(存在轉換后構不成一棵樹的情況,需要新增一個虛點作為根節點),
“笛卡爾樹上往左兒子走的次數≤m?1\leq m?1≤m?1” 等價于 “記虛點的深度為0,多叉樹的深度要≤m\leq m≤m”。
問題變成統計 有n+1n+1n+1個節點,深度≤m\leq m≤m的不同形態的多叉樹棵樹
考慮樹的括號表示,再稍微改一下定義
以這棵樹為例,常規的括號表示法為1(2(4(8,9),5(10,11)),3(6(12),7)),我把它改成
從左到右遍歷多叉樹對應的括號序列(不算最外層根代表的那對括號),要求任何時候 左括號數量≥右括號數量左括號數量\geq 右括號數量左括號數量≥右括號數量(括號序列性質),左括號數量?右括號數量≤m左括號數量-右括號數量\leq m左括號數量?右括號數量≤m(樹的深度≤m\leq m≤m)
把統計合法的括號序列個數 轉化為 統計折線數量:
從左到右遍歷括號序列,每遍歷到一個括號,xxx坐標+1,如果遍歷到的是左括號則yyy坐標+1,遍歷到的是右括號則yyy坐標-1。起點為(0,0)(0,0)(0,0),終點為(2n,0)(2n,0)(2n,0)。由于對括號序列的限制,行走過程中不能經過y=?1y=-1y=?1,y=m+1y=m+1y=m+1這兩條直線。
原問題轉為 統計從(0,0)(0,0)(0,0)走到(2n,0)(2n,0)(2n,0),每次x+1x+1x+1,y+1或?1y+1或-1y+1或?1,不經過y=?1y=-1y=?1,y=m+1y=m+1y=m+1這兩條直線的合法方案數。
先簡化問題:只限制折線不能經過y=?1y=-1y=?1,不限制其不能經過y=m+1y=m+1y=m+1
記A(x1,y1)→(x2,y2)A_{(x1,y1)\to(x2,y2)}A(x1,y1)→(x2,y2)?為從(x1,y1)(x1,y1)(x1,y1)走到(x2,y2)(x2,y2)(x2,y2),不考慮是否經過某些直線的方案數,
B(x1,y1)→(x2,y2)B_{(x1,y1)\to(x2,y2)}B(x1,y1)→(x2,y2)?為從(x1,y1)(x1,y1)(x1,y1)走到(x2,y2)(x2,y2)(x2,y2),經過y=?1y=-1y=?1的方案數,
C(x1,y1)→(x2,?1)C_{(x1,y1)\to(x2,-1)}C(x1,y1)→(x2,?1)?為從(x1,y1)(x1,y1)(x1,y1)走到(x2,?1)(x2,-1)(x2,?1),在(x2,?1)(x2,-1)(x2,?1)第一次經過y=?1y=-1y=?1的方案數。
A(x1,y1)→(x2,y2)=Cx2?x1x2?x1+y2?y12A_{(x1,y1)\to (x2,y2)}=C_{x2-x1}^{\frac{x2-x1+y2-y1}{2}}A(x1,y1)→(x2,y2)?=Cx2?x12x2?x1+y2?y1??(相當于在x2?x1x2-x1x2?x1步中選x2?x1+y2?y12\frac{x2-x1+y2-y1}{2}2x2?x1+y2?y1?步令yyy坐標+1)
考慮容斥,簡化版問題的答案為A(0,0)→(2n,0)?B(0,0)→(2n,0)A_{(0,0)\to(2n,0)}-B_{(0,0)\to(2n,0)}A(0,0)→(2n,0)??B(0,0)→(2n,0)?
B(0,0)→(2n,0)=∑p=12n?1C(0,0)→(p,?1)×A(p,?1)→(2n,0)B_{(0,0)\to(2n,0)}=\sum_{p=1}^{2n-1}C_{(0,0)\to(p,-1)}\times A_{(p,-1)\to(2n,0)}B(0,0)→(2n,0)?=p=1∑2n?1?C(0,0)→(p,?1)?×A(p,?1)→(2n,0)?
設s′s's′與s(0,0)s(0,0)s(0,0)關于y=?1y=-1y=?1對稱,根據對稱性C(0,0)→(p,?1)=Cs′→(p,?1)C_{(0,0)\to(p,-1)}=C_{s'\to (p,-1)}C(0,0)→(p,?1)?=Cs′→(p,?1)?
所以
B(0,0)→(2n,0)=∑p=12n?1Cs′→(p,?1)×A(p,?1)→(2n,0)B_{(0,0)\to(2n,0)}=\sum_{p=1}^{2n-1}C_{s'\to (p,-1)}\times A_{(p,-1)\to(2n,0)}B(0,0)→(2n,0)?=p=1∑2n?1?Cs′→(p,?1)?×A(p,?1)→(2n,0)?
等式右側那坨式子的組合意義就是As′→(2n,0)A_{s'\to(2n,0)}As′→(2n,0)?
所以
B(0,0)→(2n,0)=As′→(2n,0)B_{(0,0)\to(2n,0)}=A_{s'\to(2n,0)}B(0,0)→(2n,0)?=As′→(2n,0)?
再考慮簡化前的問題怎么做:
若要計算 “經過若干次y=m+1y=m+1y=m+1,再經過若干次y=?1y=?1y=?1,再經過若干次y=m+1y=m+1y=m+1,再.........” 的折線數量,則把起點(0,0)(0,0)(0,0)關于y=m+1y=m+1y=m+1對稱,再關于y=?1y=?1y=?1對稱,再關于y=m+1y=m+1y=m+1對稱,再......... 所求折線數量即為 最終得到的起點到(2n,0)(2n,0)(2n,0),不考慮是否經過某些直線的折線數量。
折線先經過y=?1y=-1y=?1的情況同理。
還是考慮容斥,容斥系數就是(?1)對稱次數(-1)^{對稱次數}(?1)對稱次數
參考博客
#include<bits/stdc++.h> using namespace std; typedef long long ll; const int N=1e5+10; const int mod=998244353; int n,m,fac[N<<1],ifac[N<<1]; int C(int n,int m){return (ll)fac[n]*ifac[m]%mod*ifac[n-m]%mod; } int main(){scanf("%d%d",&n,&m);if(m>n){puts("0");return 0;}fac[0]=fac[1]=ifac[0]=ifac[1]=1;for(int i=2;i<=n*2;i++) fac[i]=(ll)fac[i-1]*i%mod,ifac[i]=(ll)(mod-mod/i)*ifac[mod%i]%mod;for(int i=2;i<=n*2;i++) ifac[i]=(ll)ifac[i-1]*ifac[i]%mod;int ans=C(n*2,n);for(int i=0,x=0;;i++){if(i&1) x=-2-x;else x=(m+1)*2-x;if(abs(x)>n*2) break;if(i&1) (ans+=C(n*2,(n*2-x)/2))%=mod;else (ans+=mod-C(n*2,(n*2-x)/2))%=mod;}for(int i=0,x=0;;i++){if(i&1) x=(m+1)*2-x;else x=-2-x;if(abs(x)>n*2) break;if(i&1) (ans+=C(n*2,(n*2-x)/2))%=mod;else (ans+=mod-C(n*2,(n*2-x)/2))%=mod;}printf("%d\n", ans);return 0; } 創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的[集训队作业2018] count(笛卡尔树,生成函数,卡特兰数)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 服务器配置电脑推荐(服务器配置电脑)
- 下一篇: [POI2015] Pustynia(差