【Learning】无标号生成树计数

无标号有根树

设$f_n$为$n$个节点的无标号有根树数量。
把生成函数写出来，设$F(x)={\sum }_{i=1}^{\infty }{f}_i{x}^i$。
考虑钦点一个点作为根，则需要在下面接若干棵节点数之和为$n-1$的子树。
$$\begin{array}{rl}{f}_n& =[x^n]F(x)\\ & =[x^{n-1}]\prod _{i=1}^{\infty }(\sum _{j=0}^{\infty }{x}^{ij}{)}^{f_i}\\ & =[x^{n-1}]\prod _{i=1}^{\infty }(1-x^i{)}^{-f_i}\\ & =[x^n]x\prod _{i=1}^{\infty }(1-x^i{)}^{-f_i}\end{array}$$
上式中等比数列方案数次方的意义就是在每种大小为$i$的不同形态的树中选子树。
$$\begin{array}{r}F(x)=x\prod _{i=1}^{\infty }(1-x^i{)}^{-f_i}\end{array}$$
两边同时取$\ln$
$$\begin{array}{r}\ln F(x)=\ln x-\sum _{i=1}^{\infty }{f}_i\ln (1-x^i)\end{array}$$
两边同时求导
$$\begin{array}{r}\frac{F^{\prime }(x)}{F(x)}=\frac{1}{x}+\sum _{i=1}^{\infty }{f}_i\frac{i{x}^{i-1}}{1-x^i}\end{array}$$
化简得到
$$\begin{array}{r}x{F}^{\prime }(x)=F(x)+F(x)\sum _{i=1}^{\infty }i\cdot f_i\frac{x^i}{1-x^i}\end{array}$$
单独考虑$x^n$这一项的系数
$$\begin{array}{rl}n{f}_n& =f_n+\sum _{i=1}^{n-1}{f}_i\sum _{j=1}^{\infty }j\cdot f_j\cdot [x^{n-i}]\frac{x^j}{1-x^j}\\ & =f_n+\sum _{i=1}^{n-1}{f}_i\sum _{j=1}^{\infty }j\cdot f_j\cdot [x^{n-i}]\sum _{k=1}^{\infty }{x}^{jk}\\ & =f_n+\sum _{i=1}^{n-1}{f}_i\sum _{j=1}^{\infty }j\cdot f_j[j|n-i]\\ & =f_n+\sum _{i=1}^{n-1}{f}_i\sum _{j|n-i}j\cdot f_j\end{array}$$
化简得
$$f_n=\frac{{\sum }_{i=1}^{n-1}{f}_i{\sum }_{j|n-i}j\cdot f_j}{n-1}$$
可以$n^2$递推，也可以$n{\log }_2^{2}n$分治$FFT$。
分治的时候把贡献拆成两部分算，而且取的是多项式后一半，前一半循环卷积了也不用管。

无标号无根树

设$g_n$为$n$个节点的无标号无根树数量。
只考虑以重心为根的有根树，进行容斥。
当$n$为奇数时，
$$g_n=f_n-\sum _{i=1}^{\lfloor \frac{n}{2}\rfloor }{f}_i{f}_{n-i}$$
若$n$为偶数，则一棵树有两个重心，同一棵树有两种方法拼出来，需要特殊处理一下。
当$n$为偶数时，
$$\begin{array}{rl}{g}_n& =f_n-(\sum _{i=1}^{\frac{n}{2}-1}{f}_i{f}_{n-i})-\frac{f_{\frac{n}{2}}(f_{\frac{n}{2}}-1)}{2}\\ & =f_n-(\sum _{i=1}^{\frac{n}{2}}{f}_i{f}_{n-i})+\frac{f_{\frac{n}{2}}(f_{\frac{n}{2}}+1)}{2}\end{array}$$

简单代码

for(int i=1;i<=n;i++){	
	if(i==1){
		f[i]=1;
	}else{
		for(int j=1;j<i;j++){
			f[i]=(f[i]+1LL*f[j]*g[i-j])%mod;
		}
		f[i]=1LL*f[i]*fastpow(i-1,mod-2)%mod;
	}
	int tmp=1LL*i*f[i]%mod;
	for(int j=i;j<=n;j+=i){
		g[j]=(g[j]+tmp)%mod;
	}
}

待填坑：时间复杂度为$nlo{g}_{2}n$的多项式$\ln ,\exp$牛顿迭代做法。

$nlo{g}_2^{2}n$分治做法

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=530005,mod=998244353;
int n,len,res,rev[N],f[N],g[N],a[N],b[N];
inline int add(int a,int b){
	a+=b;
	return a<mod?a:a-mod;
}
inline int dec(int a,int b){
	a-=b;
	return a<0?a+mod:a;
}
inline int mul(int a,int b){
	return 1LL*a*b%mod;
}
int fastpow(int a,int x){
	int res=1;
	while(x){
		if(x&1){
			res=mul(res,a);
		}
		x>>=1;
		a=mul(a,a);
	}
	return res;
}
void ntt(int a[],int len,int dft){
	for(int i=0;i<len;i++){
		rev[i]=(rev[i>>1]>>1)|((i&1)*(len>>1));
		if(i<rev[i]){
			swap(a[i],a[rev[i]]);
		}
	}
	for(int i=1;i<len;i<<=1){
		int wn=fastpow(3,(mod-1)/i/2);
		if(dft==-1){
			wn=fastpow(wn,mod-2);
		}
		for(int j=0;j<len;j+=(i<<1)){
			int w=1,x,y;
			for(int k=j;k<j+i;k++,w=mul(w,wn)){
				x=a[k];
				y=mul(w,a[k+i]);
				a[k]=add(x,y);
				a[k+i]=dec(x,y);
			}
		}
	}
	if(dft==-1){
		int inv=fastpow(len,mod-2);
		for(int i=0;i<len;i++){
			a[i]=mul(a[i],inv);
		}
	}
}
void solve(int l,int r,int len){
	if(l==r){
		f[l]+=(l==1);
		if(l>1){
			f[l]=mul(f[l],fastpow(l-1,mod-2));
		}
		int t=mul(f[l],l);
		for(int i=l;i<=n&&l;i+=l){
			g[i]=add(g[i],t);
		}
		return;
	}
	solve(l,l+(len>>1)-1,len>>1);
	for(int i=0;i<len;i++){
		a[i]=(i<(len>>1))?f[l+i]:0;
		b[i]=(i<l+(len>>1))?g[i]:0;
	}
	ntt(a,len,1);
	ntt(b,len,1);
	for(int i=0;i<len;i++){
		a[i]=mul(a[i],b[i]);
	}
	ntt(a,len,-1);
	for(int i=(len>>1);i<len;i++){
		f[l+i]=add(f[l+i],a[i]);
	}
	for(int i=0;i<len;i++){
		a[i]=(i<(len>>1))?g[l+i]:0;
		b[i]=(i<l)?f[i]:0;
	}
	ntt(a,len,1);
	ntt(b,len,1);
	for(int i=0;i<len;i++){
		a[i]=mul(a[i],b[i]);
	}
	ntt(a,len,-1);
	for(int i=(len>>1);i<len;i++){
		f[l+i]=add(f[l+i],a[i]);
	}
	solve(l+(len>>1),r,len>>1);
}
int main(){
	scanf("%d",&n);
	for(len=1;len<=n;len<<=1);
	solve(0,len-1,len);
	res=f[n];
	for(int i=1;(i<<1)<=n;i++){
		res=dec(res,mul(f[i],f[n-i]));
	}
	if(~n&1){
		res=add(res,1LL*f[n>>1]*(f[n>>1]+1)/2%mod);
	}
	printf("%d\n",res);
	return 0;
}