tree 解题报告

tree

对于 \(n\) 个点带标号的无根森林，计算所有森林的树的个数的 \(k\) 次方，对 \(998244353\) 取模。

---

自闭，错了一堆关于长度的问题，这里以后一定要注意

比如需要 \(n\) 次多项式的点值，但是却使用了乘法后的 \(2\times n\) 多项式的前 \(n\) 项点值；对 \(2\times n\) 多项式按长度 \(2\times n\) DFT 直接什么的

以及还是容易写错 exp 板子的问题...

---

由 \(prufer\) 序列，我们知道树的生成函数是 \(A(x)=\sum_{i\ge 0}\frac{i^{i-2}}{i!}x^i\) ，注意这里我们要算的话需要定义 \(0\) 个点的树个数是 \(0\)

设森林个数的 \(k\) 次方的生成函数是 \(B_k(x)\) ，我们有

\[B_k(x)=\sum_{i\ge 0}i^k\frac{A^i(x)}{i!}
\]

我们可以知道 \(B_0=e^A\)

因为 \(k\) 比较小，并且我们知道 \(B_0\) ，所以考虑建立 \(B_k\) 与 \(B_{k-1}\) 之间的递推关系式

因为关键在于 \(i^k\) 与 \(i^{k-1}\) 不一样，所以考虑求导去凑

\[\begin{aligned}
B_{k-1}'(x)&=\sum_{i\ge 0}\frac{i^{k-1}}{i!}(A^i(x))'\\
&=\sum_{i\ge 0}\frac{i^k}{i!}A^{i-1}(x)A'(x)\\
&=\frac{A'(x)}{A(x)}\sum_{i\ge 0}\frac{i^k}{i!}A^i(x)\\
&=\frac{A'(x)}{A(x)}B_k(x)
\end{aligned}
\]

因此

\[B_k(x)=\frac{A(x)B'_{k-1}(x)}{A'(x)}
\]

然后套上多项式板子即可

复杂度 \(O(kn\log n)\)

---

Code:

#include <cstdio>
#include <cctype>
#include <algorithm>
using std::min;
const int SIZE=1<<21;
char ibuf[SIZE],*iS,*iT;
//#define gc() (iS==iT?(iT=(iS=ibuf)+fread(ibuf,1,SIZE,stdin),iS==iT?EOF:*iS++):*iS++)
#define gc() getchar()
template <class T>
void read(T &x)
{
	int f=0;x=0;char c=gc();
	while(!isdigit(c)) f|=c=='-',c=gc();
	while(isdigit(c)) x=x*10+c-'0',c=gc();
	if(f) x=-x;
}
const int N=(1<<17)+10;
const int mod=998244353,G=3,Gi=332748118;
inline int add(int x,int y){return x+y>=mod?x+y-mod:x+y;}
#define mul(x,y) (1ll*(x)*(y)%mod)
int qp(int d,int k){int f=1;while(k){if(k&1)f=mul(f,d);d=mul(d,d),k>>=1;}return f;}
int turn[N],fac[N],inv[N];
void NTT(int *a,int lim,int typ)
{
	int L=-1;for(int i=1;i<lim;i<<=1) ++L;
	for(int i=0;i<lim;i++)
	{
		turn[i]=turn[i>>1]>>1|(i&1)<<L;
		if(i<turn[i]) std::swap(a[i],a[turn[i]]);
	}
	for(int le=1;le<lim;le<<=1)
	{
		int wn=qp(typ?G:Gi,(mod-1)/(le<<1));
		for(int i=0;i<lim;i+=le<<1)
		{
			int w=1;
			for(int j=i;j<i+le;j++,w=mul(w,wn))
			{
				int x=a[j],y=mul(w,a[j+le]);
				a[j]=add(x,y),a[j+le]=add(x,mod-y);
			}
		}
	}
	if(!typ)
	{
		int inv=qp(lim,mod-2);
		for(int i=0;i<lim;i++) a[i]=mul(a[i],inv);
	}
}
int iva[N],ivb[N];
void polyinv(int *a,int *b,int lim)
{
	if(lim==1){b[0]=qp(a[0],mod-2);return;}
	polyinv(a,b,lim>>1);
	for(int i=0;i<lim<<1;i++) iva[i]=ivb[i]=0;
	for(int i=0;i<lim;i++) iva[i]=a[i],ivb[i]=b[i];
	NTT(iva,lim<<1,1),NTT(ivb,lim<<1,1);
	for(int i=0;i<lim<<1;i++) iva[i]=mul(ivb[i],add(2,mod-mul(iva[i],ivb[i])));
	NTT(iva,lim<<1,0);
	for(int i=0;i<lim;i++) b[i]=iva[i];
}
void polyqd(int *a,int lim)
{
	for(int i=0;i<lim-1;i++) a[i]=mul(a[i+1],i+1);
	a[lim-1]=0;
}
void polyint(int *a,int lim)
{
	for(int i=lim-1;i;i--)
        a[i]=mul(a[i-1],mul(fac[i-1],inv[i]));
	a[0]=0;
}
int lna[N],lnb[N];
void polyln(int *a,int lim)
{
	for(int i=0;i<lim<<1;i++) lna[i]=lnb[i]=0;
	for(int i=0;i<lim;i++) lna[i]=a[i];
	polyinv(lna,lnb,lim);
	polyqd(lna,lim);
	NTT(lna,lim<<1,1),NTT(lnb,lim<<1,1);
	for(int i=0;i<lim<<1;i++) lna[i]=mul(lna[i],lnb[i]);
	NTT(lna,lim<<1,0);
	polyint(lna,lim);
	for(int i=0;i<lim;i++) a[i]=lna[i];
}
int exa[N],exb[N];
void polyexp(int *a,int *b,int lim)
{
	if(lim==1){b[0]=1;return;}
	polyexp(a,b,lim>>1);
	for(int i=0;i<lim<<1;i++) exa[i]=exb[i]=0;
	for(int i=0;i<lim;i++) exa[i]=exb[i]=b[i];
	polyln(exb,lim);
	for(int i=0;i<lim;i++) exb[i]=add(a[i]+(i==0),mod-exb[i]);
	NTT(exa,lim<<1,1),NTT(exb,lim<<1,1);
	for(int i=0;i<lim<<1;i++) exa[i]=mul(exa[i],exb[i]);
	NTT(exa,lim<<1,0);
	for(int i=0;i<lim;i++) b[i]=exa[i];
}
int A[N],B[21][N],C[N],D[N];
int main()
{
	int n,k;
	read(n),read(k);
	int lim=1;
	while(lim<=n) lim<<=1;
	fac[0]=1;for(int i=1;i<=lim<<1;i++) fac[i]=mul(fac[i-1],i);
	inv[lim<<1]=qp(fac[lim<<1],mod-2);
	for(int i=(lim<<1)-1;~i;i--) inv[i]=mul(inv[i+1],i+1);
	A[0]=C[0]=0;A[1]=C[1]=1;
	for(int i=2;i<lim;i++) A[i]=C[i]=mul(qp(i,i-2),inv[i]);

	polyexp(A,B[0],lim);
	polyqd(C,lim);
	polyinv(C,D,lim);

	NTT(A,lim<<1,1),NTT(D,lim<<1,1);
	for(int i=0;i<lim<<1;i++) A[i]=mul(A[i],D[i]);

	NTT(A,lim<<1,0);//warning
	for(int i=lim;i<lim<<1;i++) A[i]=0;
	NTT(A,lim<<1,1);

	for(int i=1;i<=k;i++)
	{
		polyqd(B[i-1],lim);
		for(int j=lim;j<lim<<1;j++) B[i-1][j]=0;//warning
		NTT(B[i-1],lim<<1,1);
		for(int j=0;j<lim<<1;j++) B[i][j]=mul(B[i-1][j],A[j]);
		NTT(B[i],lim<<1,0);
	}
	printf("%lld\n",mul(B[k][n],fac[n]));
	return 0;
}

---

2019.6.24