这种题想清楚之后基本上就是乱写都能AC。
做集训队作业,第一步永远都是模型转换。。。
发现除了最后一个操作,其他操作产生的字符一定会被一个1操作消掉。
可以转化成有根树上的父子关系。
如果一个操作 a a a被另一个操作 b b b消除,则我们令 b b b为 a a a父亲。问题转化为满足下列条件的树的计数:
我们在 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=[n−1∈B]+i∈A∑i!1k=1∑k!1∑j=1kaj=n−1−i,aj≥2∑∏j=1kaj!(n−1)!j=1∏kfaj
边界 f 0 = 0 , f 1 = 1 f_0=0,f_1=1 f0=0,f1=1,因为我们现在保证了0在集合 B B B中。
好我承认这个式子很长。。。
两边同时除掉 ( n − 1 ) ! (n-1)! (n−1)!会比较容易看出来:
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!} (n−1)!fn=(n−1)![n−1∈B]+i∑i![i∈A]k=1∑k!1∑j=1kaj=n−1−i,aj≥2∑j=1∏kaj!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(F−x)−1)F′=Ae−xeF+(B−A)
设 C = A e − x , D = B − A C=Ae^{-x},D=B-A C=Ae−x,D=B−A,则:
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} F′≡h(F)(modxn)
假设现在我们已经解得了 F 0 ′ ≡ h ( F 0 ) ( m o d x n 2 ) F_0'\equiv h(F_0)\pmod {x^\frac{n}{2}} F0′≡h(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} F′≡h(F0)+h′(F0)F−h′(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} F′vF′v−h′(F0)vFF′v+v′F(Fv)′F=v(h(F0)+h′(F0)(F−F0))=v(h(F0)−h′(F0)F0)=v(h(F0)−h′(F0)F0)=v(h(F0)−h′(F0)F0)=v1∫v(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+D,h′(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=v1∫v(CeF0(1−F0)+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;
}