CVBB ACM Team

到此差别页面的链接

--- 2020-2021:teams:legal_string:jxm2001:生成函数_1 [2020/08/09 15:21]
jxm2001
+++ 2020-2021:teams:legal_string:jxm2001:生成函数_1 [2020/08/21 20:48] (当前版本)
jxm2001
@@ 行 1: / 行 1: @@
-====== 普通生成函数 ======
+====== 普通生成函数(OGF) ======
 =====  算法简介 =====
-形如 $F(x)=\sum_{n=0}^{\infty}a_nx^n$ 的函数， $a_n$ 可以提供关于这个序列的信息，一般用于解决组合计算问题。
+形如 $F(x)=\sum_{n=0}^{\infty}a_nx^n$ 的函数， $a_n$ 可以提供关于这个序列的信息，一般用于解决无标号组合计数问题。
 ===== 算法例题 =====
@@ 行 138: / 行 138: @@
 $$
-于是 $p_n=[n==0]+p_{n-1}+p{n-2}-p_{n-5}-p_{n-7}+\cdots$。
+于是 $p_n=[n==0]+p_{n-1}+p_{n-2}-p_{n-5}-p_{n-7}+\cdots$。
 ==== 例题四 ====
+[[https://www.luogu.com.cn/problem/P4389|洛谷p4389]]
+=== 题意 ===
+给定 $n$ 种物品，每种物品体积为 $v_i$，有无限个。问物品恰好装满体积为 $i(1\le i\le m)$ 的背包的方案数。
+=== 题解 ===
+首先只选每个物品 $i$ 的生成函数为 $1+x^{v_i}+x^{2v_i}+\cdots=\cfrac 1{1-x^{v_i}}$。
+于是最终方案的生成函数为 $F(x)=\prod_{i=1}^n \cfrac 1{1-x^{v_i}}=\prod_{i=1}^n \sum_{j=1}^{\infty}\cfrac {x^{jv_i}}j$。
+记体积为 $i$ 的物品有 $c_i$ 个，同时考虑取 $\ln$ 加速乘法，有 $F(x)=\exp \ln F(x)=\exp (\sum_{i=1}^n \sum_{j=1}^{\infty}\cfrac {x^{jv_i}}j)=\exp (\sum_{i=1}^n \sum_{j=1}^{\infty}\cfrac {x^{ij}}j)$。
+由于只需要计算 $[x^i]F(x)(1\le i\le m)$，于是只需要计算出 $[x^i](\sum_{i=1}^n \sum_{j=1}^{\infty}\cfrac {x^{ij}}j)(0\le i\le m)$ 即可。
+时间复杂度 $O(m\log m)$。
+<hidden 查看代码>
+<code cpp>
+const int MAXN=1e5+5,Mod=998244353;
+int quick_pow(int a,int b){
+	int ans=1;
+	while(b){
+		if(b&1)
+		ans=1LL*ans*a%Mod;
+		a=1LL*a*a%Mod;
+		b>>=1;
+	}
+	return ans;
+}
+namespace Poly{
+	const int G=3;
+	int rev[MAXN<<2],Wn[30][2];
+	void init(){
+		int m=Mod-1,lg2=0;
+		while(m%2==0)m>>=1,lg2++;
+		Wn[lg2][1]=quick_pow(G,m);
+		Wn[lg2][0]=quick_pow(Wn[lg2][1],Mod-2);
+		while(lg2){
+			m<<=1,lg2--;
+			Wn[lg2][0]=1LL*Wn[lg2+1][0]*Wn[lg2+1][0]%Mod;
+			Wn[lg2][1]=1LL*Wn[lg2+1][1]*Wn[lg2+1][1]%Mod;
+		}
+	}
+	int build(int k){
+		int n,pos=0;
+		while((1<<pos)<=k)pos++;
+		n=1<<pos;
+		_for(i,0,n)rev[i]=(rev[i>>1]>>1)|((i&1)<<(pos-1));
+		return n;
+	}
+	void NTT(int *f,int n,bool type){
+		_for(i,0,n)if(i<rev[i])
+		swap(f[i],f[rev[i]]);
+		int t1,t2;
+		for(int i=1,lg2=0;i<n;i<<=1,lg2++){
+			int w=Wn[lg2+1][type];
+			for(int j=0;j<n;j+=(i<<1)){
+				int cur=1;
+				_for(k,j,j+i){
+					t1=f[k],t2=1LL*cur*f[k+i]%Mod;
+					f[k]=(t1+t2)%Mod,f[k+i]=(t1-t2)%Mod;
+					cur=1LL*cur*w%Mod;
+				}
+			}
+		}
+		if(!type){
+			int div=quick_pow(n,Mod-2);
+			_for(i,0,n)f[i]=(1LL*f[i]*div%Mod+Mod)%Mod;
+		}
+	}
+	void Mul(int *f,int _n,int *g,int _m,int xmod=0){
+		int n=build(_n+_m-2);
+		_for(i,_n,n)f[i]=0;_for(i,_m,n)g[i]=0;
+		NTT(f,n,true);NTT(g,n,true);
+		_for(i,0,n)f[i]=1LL*f[i]*g[i]%Mod;
+		NTT(f,n,false);
+		if(xmod)_for(i,xmod,n)f[i]=0;
+	}
+	void Inv(const int *f,int *g,int _n){
+		static int temp[MAXN<<2];
+		if(_n==1)return g[0]=quick_pow(f[0],Mod-2),void();
+		Inv(f,g,(_n+1)>>1);
+		int n=build((_n-1)<<1);
+		_for(i,(_n+1)>>1,n)g[i]=0;
+		_for(i,0,_n)temp[i]=f[i];_for(i,_n,n)temp[i]=0;
+		NTT(g,n,true);NTT(temp,n,true);
+		_for(i,0,n)g[i]=(2-1LL*temp[i]*g[i]%Mod)*g[i]%Mod;
+		NTT(g,n,false);
+		_for(i,_n,n)g[i]=0;
+	}
+	void Ln(const int *f,int *g,int _n){
+		static int temp[MAXN<<2];
+		Inv(f,g,_n);
+		_for(i,1,_n)temp[i-1]=1LL*f[i]*i%Mod;
+		temp[_n-1]=0;
+		Mul(g,_n,temp,_n-1,_n);
+		for(int i=_n-1;i;i--)g[i]=1LL*g[i-1]*quick_pow(i,Mod-2)%Mod;
+		g[0]=0;
+	}
+	void Exp(const int *f,int *g,int _n){
+		static int temp[MAXN<<2];
+		if(_n==1)return g[0]=1,void();
+		Exp(f,g,(_n+1)>>1);
+		_for(i,(_n+1)>>1,_n)g[i]=0;
+		Ln(g,temp,_n);
+		temp[0]=(1+f[0]-temp[0])%Mod;
+		_for(i,1,_n)temp[i]=(f[i]-temp[i])%Mod;
+		Mul(g,(_n+1)>>1,temp,_n,_n);
+	}
+}
+int a[MAXN<<2],b[MAXN<<2],c[MAXN],inv[MAXN];
+void get_inv(){
+	inv[1]=1;
+	_for(i,2,MAXN)
+	inv[i]=1LL*(Mod-Mod/i)*inv[Mod%i]%Mod;
+}
+int main()
+{
+	Poly::init();
+	get_inv();
+	int n=read_int(),m=read_int();
+	_for(i,0,n)c[read_int()]++;
+	_rep(i,1,m){
+		for(int j=1;i*j<=m;j++)
+		a[i*j]=(a[i*j]+1LL*c[i]*inv[j])%Mod;
+	}
+	Poly::Exp(a,b,m+1);
+	_rep(i,1,m)enter(b[i]);
+	return 0;
+}
+</code>
+</hidden>
+==== 例题五 ====
+[[https://www.luogu.com.cn/problem/P4451|洛谷p4451]]
+=== 题意 ===
+求 $\sum\prod_{i=1}^mF_{a_i}$，其中
+  - $m\gt 0$
+  - $a_i\gt 0$
+  - $sum_{i=1}^m a_i=n$
+  - $F$ 为斐波那契数列，且 $F_0=0,F_1=1$
+答案对 $10^9+7$ 取模。
+=== 题解 ===
+先考虑 $m=1$ 的生成函数，有 $G(x)=\sum_{i=0}^n F_ix^i$。
+于是答案的生成函数为 $H(x)=\sum_{m=1}^{\infty} G^m(x)=\cfrac {G(x)}{1-G(x)}$，接下来考虑求解 $G(x)$。
+有 $G(x)-xG(x)-x^2G(x)=F_0+F_1x-F_0=x$，于是 $G(x)=\cfrac x{1+x+x^2}$，代入可得 $H(x)=\cfrac {x}{1-2x-x^2}$。
+于是有 $(1-2x-x^2)H(x)=x$，于是 $h_n=2h_{n-1}+h_{n-2}+[n==1]$。
+通过特征根求解得 $h_n=\cfrac {\sqrt 2}4(1+\sqrt 2)^n-\cfrac {\sqrt 2}4(1-\sqrt 2)^n$。
+<hidden 查看代码>
+<code cpp>
+const int MAXL=1e4+5,Mod=1e9+7,sqrt2=59713600,inv4=250000002;
+int quick_pow(int a,int b){
+	int ans=1;
+	while(b){
+		if(b&1)ans=1LL*ans*a%Mod;
+		a=1LL*a*a%Mod;
+		b>>=1;
+	}
+	return ans;
+}
+char buf[MAXL];
+int main()
+{
+	scanf("%s",buf);
+	int n=0,len=strlen(buf);
+	_for(i,0,len)
+	n=(10LL*n+buf[i]-'0')%(Mod-1);
+	int ans=1LL*(quick_pow(1+sqrt2,n)-quick_pow(1-sqrt2,n))*sqrt2%Mod*inv4%Mod;
+	enter((ans+Mod)%Mod);
+	return 0;
+}
+</code>
+</hidden>
+==== 例题六 ====
 [[https://ac.nowcoder.com/acm/contest/5670/C|牛客暑期多校(第五场) C 题]]
@@ 行 146: / 行 336: @@
 === 题意 ===
-已知 $\sum_{i=1}^k a_i=n,\sum_{i=1}^k b_i=m,P=\sum_{i=1}^k \min(a_i,b_i)$，$\sum_{a,b}P$。
+已知 $\sum_{i=1}^k a_i=n,\sum_{i=1}^k b_i=m,P=\prod_{i=1}^k \min(a_i,b_i)$，求 $\sum_{a,b}P$。
 === 题解 ===
-令 $F(x)=\sum_{i=1,j=1}^{\infty} x^iy_j$，于是答案为 $[x^ny^m]F^k(x)$。
+令 $F(x)=\sum_{i=1,j=1}^{\infty} \min(a_i,b_i)x^iy^j$，于是答案为 $[x^ny^m]F^k(x)$。
 考虑求出 $F(x)$ 的封闭式。
@@ 行 189: / 行 379: @@
 $$
-考虑枚举 $c$ 的同时计算出 $a,b$ 即可，时间复杂度 $O(n)$。
+考虑枚举 $c$ 的同时计算出 $a,b$ 即可，于是 $[x^ny^m]F^k(x)=\sum_{i=0}^{\min(n,m)-k}{n-i-1 \choose n-k-i}{m-i-1 \choose m-k-i}{k+i-1 \choose i}$
+预处理阶乘和阶乘的逆即可，时间复杂度 $O(n)$。
+<hidden 查看代码>
+<code cpp>
+const int MAXN=1e6+5,Mod=998244353;
+int frac[MAXN],invfrac[MAXN];
+int quick_pow(int a,int b){
+	int ans=1;
+	while(b){
+		if(b&1)
+		ans=1LL*ans*a%Mod;
+		a=1LL*a*a%Mod;
+		b>>=1;
+	}
+	return ans;
+}
+int C(int n,int m){return 1LL*frac[n]*invfrac[n-m]%Mod*invfrac[m]%Mod;}
+int main()
+{
+	frac[0]=1;
+	_for(i,1,MAXN)frac[i]=1LL*frac[i-1]*i%Mod;
+	invfrac[MAXN-1]=quick_pow(frac[MAXN-1],Mod-2);
+	for(int i=MAXN-1;i;i--)
+	invfrac[i-1]=1LL*invfrac[i]*i%Mod;
+	int T=read_int();
+	while(T--){
+		int n=read_int(),m=read_int(),k=read_int();
+		int K=min(n,m)-k,ans=0;
+		_rep(i,0,K)
+		ans=(ans+1LL*C(n-i-1,n-k-i)*C(m-i-1,m-k-i)%Mod*C(k+i-1,i))%Mod;
+		enter(ans);
+	}
+	return 0;
+}
+</code>
+</hidden>
+==== 例题七 ====
+[[https://www.luogu.com.cn/problem/CF923E|CF923E]]
+=== 题意 ===
+给定一个数 $x\in [0,n]$，其中 $x=i$ 的概率为 $p_i$。每次操作将 $x$ 等概率变成 $[0,x]$ 中的某个数。问 $m$ 轮操作后 $x=i$ 的概率。
+=== 题解 ===
+设 $f_{k,i}$ 表示 $k$ 轮操作后 $x=k$ 的概率，显然有 $f_{0,i}=p_i$，并可以得到递推式
+$$f_{k,i}=\sum_{j=i}^{n} \frac {f_{k-1,j}}{j+1}$$
+考虑构造生成函数优化递推过程
+$$
+\begin{equation}\begin{split}
+F_k(x)&=\sum_{i=0}^n f_{k,i}x^i\\
+&=\sum_{i=0}^n \sum_{j=i}^{n} \frac {f_{k-1,j}}{j+1}x^i\\
+&=\sum_{j=0}^n \frac {f_{k-1,j}}{j+1}\sum_{i=0}^j x^i\\
+&=\frac 1{x-1}\sum_{j=0}^n f_{k-1,j}\frac {x^{j+1}-1}{j+1}\\
+&=\frac 1{x-1}\sum_{j=0}^n f_{k-1,j}\int_1^x t^j \mathrm{d}t\\
+&=\frac 1{x-1}\int_1^x F_{k-1}(t)\mathrm{d}t
+\end{split}\end{equation}
+$$
+由于 $\frac 1{x-1}$ 和 $\int_1^x$ 不利于更进一步处理，于是考虑构造辅助函数 $G_k(x)=F_k(x+1)$
+$$
+\begin{equation}\begin{split}
+G_k(x)&=\frac 1x\int_1^{x+1} F_{k-1}(t)\mathrm{d}t\\
+&=\frac 1x\int_0^{x} F_{k-1}(t+1)\mathrm{d}t\\
+&=\frac 1x\int_0^{x} G_{k-1}(t)\mathrm{d}t\\
+&=\sum_{i=0}^n \frac {g_{k-1,i}}{i+1}x^i
+\end{split}\end{equation}
+$$
+于是有 $g_{k,i}=\frac {g_{k-1,i}}{i+1}$，所以 $g_{m,i}=\frac {g_{0,i}}{(i+1)^m}$。
+接下来考虑 $g_{k,i}$ 与 $f_{k,i}$ 之间的转化，简记 $g_i=g_{k,i},f_i=f_{k,i}$，于是有
+$$g_i=\sum_{j=i}^n {j\choose i}f_j=\sum_{j=i}^n \frac{j!}{i!(j-i)!}f_j=\frac 1{i!}\sum_{j=0}^{n-i}\frac {(i+j)!f_{i+j}}{j!}$$
+记 $a_i=\frac 1{i!},b_i=(n-i)!f_{n-i}$，于是有 $g_i=\sum_{j=0}^{n-i}a_ib_{n-i-j}$。
+直接 $\text{NTT}$ 可以由 $F_0(x)$ 求出 $G_0(x)$ 进而求出 $G_m(x)$。考虑对 $\sum_{i=0}^n \frac 1{i!}x^i$ 求逆再与 $G_m(x)$ 卷积即可求出 $F_m(x)$，总时间复杂度 $O(n\log n)$。
+<hidden 查看代码>
+<code cpp>
+const int MAXN=1e5+5,Mod=998244353;
+int quick_pow(int a,int b){
+	int ans=1;
+	while(b){
+		if(b&1)
+		ans=1LL*ans*a%Mod;
+		a=1LL*a*a%Mod;
+		b>>=1;
+	}
+	return ans;
+}
+namespace Poly{
+	const int G=3;
+	int rev[MAXN<<2],Wn[30][2];
+	void init(){
+		int m=Mod-1,lg2=0;
+		while(m%2==0)m>>=1,lg2++;
+		Wn[lg2][1]=quick_pow(G,m);
+		Wn[lg2][0]=quick_pow(Wn[lg2][1],Mod-2);
+		while(lg2){
+			m<<=1,lg2--;
+			Wn[lg2][0]=1LL*Wn[lg2+1][0]*Wn[lg2+1][0]%Mod;
+			Wn[lg2][1]=1LL*Wn[lg2+1][1]*Wn[lg2+1][1]%Mod;
+		}
+	}
+	int build(int k){
+		int n,pos=0;
+		while((1<<pos)<=k)pos++;
+		n=1<<pos;
+		_for(i,0,n)rev[i]=(rev[i>>1]>>1)|((i&1)<<(pos-1));
+		return n;
+	}
+	void NTT(int *f,int n,bool type){
+		_for(i,0,n)if(i<rev[i])
+		swap(f[i],f[rev[i]]);
+		int t1,t2;
+		for(int i=1,lg2=0;i<n;i<<=1,lg2++){
+			int w=Wn[lg2+1][type];
+			for(int j=0;j<n;j+=(i<<1)){
+				int cur=1;
+				_for(k,j,j+i){
+					t1=f[k],t2=1LL*cur*f[k+i]%Mod;
+					f[k]=(t1+t2)%Mod,f[k+i]=(t1-t2)%Mod;
+					cur=1LL*cur*w%Mod;
+				}
+			}
+		}
+		if(!type){
+			int div=quick_pow(n,Mod-2);
+			_for(i,0,n)f[i]=(1LL*f[i]*div%Mod+Mod)%Mod;
+		}
+	}
+	void Mul(int *f,int _n,int *g,int _m,int xmod=0){
+		int n=build(_n+_m-2);
+		_for(i,_n,n)f[i]=0;_for(i,_m,n)g[i]=0;
+		NTT(f,n,true);NTT(g,n,true);
+		_for(i,0,n)f[i]=1LL*f[i]*g[i]%Mod;
+		NTT(f,n,false);
+		if(xmod)_for(i,xmod,n)f[i]=0;
+	}
+	void Inv(const int *f,int *g,int _n){
+		static int temp[MAXN<<2];
+		if(_n==1)return g[0]=quick_pow(f[0],Mod-2),void();
+		Inv(f,g,(_n+1)>>1);
+		int n=build((_n-1)<<1);
+		_for(i,(_n+1)>>1,n)g[i]=0;
+		_for(i,0,_n)temp[i]=f[i];_for(i,_n,n)temp[i]=0;
+		NTT(g,n,true);NTT(temp,n,true);
+		_for(i,0,n)g[i]=(2-1LL*temp[i]*g[i]%Mod)*g[i]%Mod;
+		NTT(g,n,false);
+		_for(i,_n,n)g[i]=0;
+	}
+}
+int a[MAXN<<2],b[MAXN<<2],c[MAXN<<2],frac[MAXN],invfrac[MAXN];
+int main()
+{
+	Poly::init();
+	frac[0]=1;
+	_for(i,1,MAXN)frac[i]=1LL*frac[i-1]*i%Mod;
+	invfrac[MAXN-1]=quick_pow(frac[MAXN-1],Mod-2);
+	for(int i=MAXN-1;i;i--)invfrac[i-1]=1LL*invfrac[i]*i%Mod;
+	int n=read_int(),m=read_LL()%(Mod-1)*(Mod-2)%(Mod-1);
+	_rep(i,0,n)
+	a[n-i]=1LL*read_int()*frac[i]%Mod;
+	_rep(i,0,n)
+	b[i]=invfrac[i];
+	Poly::Mul(a,n+1,b,n+1);
+	_rep(i,0,n)
+	a[i]=1LL*a[i]*quick_pow(n-i+1,m)%Mod,b[i]=invfrac[i];
+	Poly::Inv(b,c,n+1);
+	Poly::Mul(a,n+1,c,n+1);
+	_rep(i,0,n)a[i]=1LL*a[i]*invfrac[n-i]%Mod;
+	reverse(a,a+n+1);
+	_rep(i,0,n)space(a[i]);
+	return 0;
+}
+</code>
+</hidden>

CVBB ACM Team

用户工具

站点工具

页面工具