【集训队作业2018】普通的计数题(牛顿迭代)(常微分方程)

传送门


这种题想清楚之后基本上就是乱写都能AC。

题解:

做集训队作业,第一步永远都是模型转换。。。

发现除了最后一个操作,其他操作产生的字符一定会被一个1操作消掉。

可以转化成有根树上的父子关系。

如果一个操作 a a a被另一个操作 b b b消除,则我们令 b b b a a a父亲。问题转化为满足下列条件的树的计数:

  1. 标号为1到n,且标号满足大根堆性质。
  2. 对于一个非叶节点,如果它有儿子是非叶节点,则它的叶子儿子个数必须在集合 A A A中,否则,它的叶子儿子个数必须在集合 B B B中。

我们在 B B B里面塞一个 0 0 0即可把上述限制转化为针对任意节点,包括叶子。

f i f_i fi表示 i i i个点的方案数,枚举儿子个数和大小分配标号可以知道:

f n = [ n − 1 ∈ B ] + ∑ i ∈ A 1 i ! ∑ k = 1 1 k ! ∑ ∑ j = 1 k a j = n − 1 − i , a j ≥ 2 ( n − 1 ) ! ∏ j = 1 k a j ! ∏ j = 1 k f a j f_n=[n-1\in B]+\sum_{i\in A}\frac{1}{i!}\sum_{k=1}\frac{1}{k!}\sum_{\sum_{j=1}^ka_j=n-1-i,a_j\geq 2}\frac{(n-1)!}{\prod_{j=1}^ka_j!}\prod_{j=1}^kf_{a_j} fn=[n1B]+iAi!1k=1k!1j=1kaj=n1i,aj2j=1kaj!(n1)!j=1kfaj

边界 f 0 = 0 , f 1 = 1 f_0=0,f_1=1 f0=0,f1=1,因为我们现在保证了0在集合 B B B中。

好我承认这个式子很长。。。

两边同时除掉 ( n − 1 ) ! (n-1)! (n1)!会比较容易看出来:

f n ( n − 1 ) ! = [ n − 1 ∈ B ] ( n − 1 ) ! + ∑ i [ i ∈ A ] i ! ∑ k = 1 1 k ! ∑ ∑ j = 1 k a j = n − 1 − i , a j ≥ 2 ∏ j = 1 k f a j a j ! \frac{f_n}{(n-1)!}=\frac{[n-1\in B]}{(n-1)!}+\sum_{i}\frac{[i\in A]}{i!}\sum_{k=1}\frac{1}{k!}\sum_{\sum_{j=1}^ka_j=n-1-i,a_j\geq 2}\prod_{j=1}^k\frac{f_{a_j}}{a_j!} (n1)!fn=(n1)![n1B]+ii![iA]k=1k!1j=1kaj=n1i,aj2j=1kaj!faj

F ( x ) , A ( x ) , B ( x ) F(x),A(x),B(x) F(x),A(x),B(x)分别是序列 f f f,集合 A A A,集合 B B B的EGF,我们得到:

F ′ = B + A ( exp ⁡ ( F − x ) − 1 ) F ′ = A e − x e F + ( B − A ) F'=B+A(\exp(F-x)-1)\\ F'=Ae^{-x}e^F+(B-A) F=B+A(exp(Fx)1)F=AexeF+(BA)

C = A e − x , D = B − A C=Ae^{-x},D=B-A C=Aex,D=BA,则:

F ′ = C e F + D F'=Ce^F+D F=CeF+D

嗯,常微分方程。

h ( t ) = C e t + D h(t)=Ce^t+D h(t)=Cet+D。我们要解决的问题就是 F ′ ≡ h ( F ) ( m o d x n ) F'\equiv h(F)\pmod {x^n} Fh(F)(modxn)

假设现在我们已经解得了 F 0 ′ ≡ h ( F 0 ) ( m o d x n 2 ) F_0'\equiv h(F_0)\pmod {x^\frac{n}{2}} F0h(F0)(modx2n),考虑怎么扩展到 m o d    x n \mod x^n modxn

首先将 h h h F 0 F_0 F0处泰勒展开,得到:

F ′ ≡ h ( F 0 ) + h ′ ( F 0 ) F − h ′ ( F 0 ) F 0 ( m o d x n ) F'\equiv h(F_0)+h'(F_0)F-h'(F_0)F_0\pmod {x^n} Fh(F0)+h(F0)Fh(F0)F0(modxn)

v ( x ) = exp ⁡ ( − ∫ h ′ ( F 0 ( x ) ) d x ) v(x)=\exp(-\int h'(F_0(x)) \mathrm{d} x) v(x)=exp(h(F0(x))dx),显然有 v ′ = − h ′ ( F 0 ) v v'=-h'(F_0)v v=h(F0)v

把上面的等式两边同时乘上 v v v,开始转化,(以下讨论全部 m o d    x n \mod x^n modxn):

F ′ v = v ( h ( F 0 ) + h ′ ( F 0 ) ( F − F 0 ) ) F ′ v − h ′ ( F 0 ) v F = v ( h ( F 0 ) − h ′ ( F 0 ) F 0 ) F ′ v + v ′ F = v ( h ( F 0 ) − h ′ ( F 0 ) F 0 ) ( F v ) ′ = v ( h ( F 0 ) − h ′ ( F 0 ) F 0 ) F = 1 v ∫ v ( h ( F 0 ) − h ′ ( F 0 ) F 0 ) d x \begin{aligned} F'v&=v(h(F_0)+h'(F_0)(F-F_0))\\ F'v-h'(F_0)vF&=v(h(F_0)-h'(F_0)F_0)\\ F'v+v'F&=v(h(F_0)-h'(F_0)F_0)\\ (Fv)'&=v(h(F_0)-h'(F_0)F_0)\\ F&=\frac{1}{v}\int v(h(F_0)-h'(F_0)F_0) \mathrm{d} x \end{aligned} FvFvh(F0)vFFv+vF(Fv)F=v(h(F0)+h(F0)(FF0))=v(h(F0)h(F0)F0)=v(h(F0)h(F0)F0)=v(h(F0)h(F0)F0)=v1v(h(F0)h(F0)F0)dx

h ( t ) = C e t + D , h ′ ( t ) = C e t h(t)=Ce^t+D,h'(t)=Ce^t h(t)=Cet+Dh(t)=Cet代入:

v = exp ⁡ ( − ∫ C e F 0 d x ) F = 1 v ∫ v ( C e F 0 ( 1 − F 0 ) + D ) d x v=\exp(-\int Ce^{F_0}\mathrm{d}x)\\ F=\frac{1}{v}\int v(Ce^{F_0}(1-F_0)+D)\mathrm{d}x v=exp(CeF0dx)F=v1v(CeF0(1F0)+D)dx

迭代即可。

下面的代码最慢的点跑了1.2s


代码:

#include
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<22|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	template<typename T>
	inline T get(){
		char c;T num;
		while(!isdigit(c=gc()));num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline int gi(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int mod=998244353;
inline int add(int a,int b){a+=b-mod;return a+(a>>31&mod);}
inline int dec(int a,int b){a-=b;return a+(a>>31&mod);}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){a+=b-mod;a+=a>>31&mod;}
inline void Dec(int &a,int b){a-=b;a+=a>>31&mod;}
inline void Mul(int &a,int b){a=mul(a,b);}

typedef std::vector<int> Poly;

cs int bit=19,SIZE=1<<bit|1;

int r[SIZE],*w[bit+1];
int fac[SIZE],ifac[SIZE],inv[SIZE];
inline void init_NTT(){
	for(int re i=1;i<=bit;++i)w[i]=new int[1<<i-1];
	int wn=power(3,mod-1>>bit);w[bit][0]=1;
	for(int re i=1;i<(1<<bit-1);++i)w[bit][i]=mul(w[bit][i-1],wn);
	for(int re i=bit-1;i;--i)
	for(int re j=0;j<(1<<i-1);++j)w[i][j]=w[i+1][j<<1];
	fac[0]=fac[1]=ifac[0]=ifac[1]=inv[0]=inv[1]=1;
	for(int re i=2;i<SIZE;++i){
		fac[i]=mul(fac[i-1],i);
		inv[i]=mul(inv[mod%i],mod-mod/i);
		ifac[i]=mul(ifac[i-1],inv[i]);
	}
}
inline void NTT(int *A,int len,int typ){
	for(int re i=1;i<len;++i)if(i<r[i])std::swap(A[i],A[r[i]]);
	for(int re i=1,d=1;i<len;i<<=1,++d)
	for(int re j=0;j<len;j+=i<<1)
	for(int re k=0;k<i;++k){
		int &t1=A[j+k],&t2=A[i+j+k],t=mul(t2,w[d][k]);
		t2=dec(t1,t),Inc(t1,t);
	}
	if(typ==-1){
		std::reverse(A+1,A+len);
		for(int re i=0,iv=inv[len];i<len;++i)Mul(A[i],iv);
	}
}
inline void NTT(Poly &A,int len,int typ){NTT(&A[0],len,typ);}
inline void init_rev(int l){
	for(int re i=1;i<l;++i)r[i]=r[i>>1]>>1|((i&1)?l>>1:0);
}

inline Poly operator*(Poly a,Poly b){
	int n=a.size(),m=b.size(),deg=n+m-1,l=1;
	while(l<deg)l<<=1;init_rev(l);
	a.resize(l),NTT(a,l,1);
	b.resize(l),NTT(b,l,1);
	for(int re i=0;i<l;++i)Mul(a[i],b[i]);
	NTT(a,l,-1);a.resize(deg);
	return a;
}

inline Poly Inv(cs Poly &a,int lim){
	int n=a.size();Poly c,b(1,power(a[0],mod-2));
	for(int re l=4;(l>>2)<lim;l<<=1){
		init_rev(l);
		c.resize(l>>1);for(int re i=0;i<(l>>1);++i)c[i]=i<n?a[i]:0;
		c.resize(l),NTT(c,l,1);
		b.resize(l),NTT(b,l,1);
		for(int re i=0;i<l;++i)Mul(b[i],dec(2,mul(b[i],c[i])));
		NTT(b,l,-1),b.resize(l>>1);
	}b.resize(lim);
	return b;
}

inline Poly Deriv(Poly a){assert(a.size());
	for(int re i=0;i+1<a.size();++i)a[i]=mul(a[i+1],i+1);
	a.pop_back();return a;
}
inline Poly Integ(Poly a){
	a.push_back(0);
	for(int re i=a.size()-1;i;--i)a[i]=mul(a[i-1],inv[i]);
	a[0]=0;return a;
}

inline Poly Ln(Poly a,int lim){
	a=Integ(Deriv(a)*Inv(a,lim));
	a.resize(lim);return a;
}

inline Poly Exp(cs Poly &a,int lim){
	int n=a.size();Poly c,b(1,1);
	for(int re i=2;(i>>1)<lim;i<<=1){
		c=Ln(b,i);
		for(int re j=0;j<i;++j)c[j]=dec(j<n?a[j]:0,c[j]);
		Inc(c[0],1);
		b=b*c;b.resize(i);
	}b.resize(lim);
	return b;
}

cs int N=1<<18|1;

int n,p,q;
int A[N],B[N],C[N],D[N];

inline void pre_work(){
	int mxa=0;
	while(p--){
		int a=gi();
		A[a]=ifac[a];
	}
	while(q--){
		int b=gi();
		B[b]=ifac[b];
	}B[0]=1;
	for(int re i=0;i<n;++i)D[i]=dec(B[i],A[i]);
	int l=1;while(l<n+n)l<<=1;init_rev(l);
	for(int re i=0;i<n;++i)C[i]=(i&1)?ifac[i]:mod-ifac[i];
	NTT(C,l,1);NTT(A,l,1);
	for(int re i=0;i<l;++i)Mul(C[i],A[i]);
	NTT(C,l,-1);
}

inline Poly Newton(int lim){
	if(lim==1)return Poly(1,0);
	Poly F0=Newton(lim+1>>1);
	Poly H=Poly(C,C+lim)*Exp(F0,lim);
	H.resize(lim);Poly v=Exp(Integ(H),lim);
	Dec(F0[0],1);H=H*F0;H.resize(lim);
	for(int re i=0;i<lim;++i)Inc(H[i],D[i]);
	Poly F=Integ(v*H);F.resize(lim);
	F=F*Inv(v,lim);F.resize(lim);
	return F;
}

signed main(){
#ifdef zxyoi
	freopen("count.in","r",stdin);
#endif
	init_NTT();
	n=gi(),p=gi(),q=gi();
	pre_work();
	Poly F=Newton(n+1);
	cout<<mul(F[n],fac[n])<<"\n";
	return 0;
}

你可能感兴趣的:(多项式,生成函数)