【BZOJ 4228】Tibbar的后花园

传送门

---

problem

求 $n$ 个带标号的点能满足如下条件的无向图个数：

1. 不存在重边和自环。
2. 不存在三个点 $a,b,c$，使三个点距离两两相等。

输出答案对 $1004535809$（$479\times 2^{21}+1$，质数）取模的结果。

数据范围：$n\le 10^6$。

---

solution

有一个结论，即满足条件的无向图，要么是一条链，要么是一个长度不为 $3$ 的倍数的环。

因为当一个点 $x$ 的度数 $\ge 3$ 时，与 $x$ 相连的 $3$ 个点之间距离两两相等，不符合题意。

我们先构造出链的 EGF，即：

$$A(x)=\sum _{i=0}^n\frac{i!}{2}\frac{x^i}{i!}=\sum _{i=0}^n\frac{x^i}{2}$$

就是长度为 $i$ 的链有 $\frac{i!}{2}$ 种。还有注意一次项的系数为 $1$。

然后是环的 EGF（注意环长不为 $3$ 的倍数）：

$$B(x)=\sum _{i=4}^n[i\%3≠0]\frac{i!}{2i}\frac{x^i}{i!}=\sum _{i=4}^n[i\%3≠0]\frac{x^i}{2i}$$

如果设整张图的 EGF 为 $G(x)=\sum _{i=0}^n{g}_i\frac{x^i}{i!}$，那么答案就是 $g_n$，即 $G(n)\times n!$。

由于整张图是 $A$ 和 $B$ 的并集拼出来的，所以 $G=e^{A+B}$，多项式 Exp 即可。

时间复杂度 $O(n\log n)$。

---

code

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define N 5000005
#define P 1004535809
using namespace std;
typedef vector<int> poly;
const int g=3;
int n,pos[N],fac[N],inv[N];
int add(int x,int y)  {return x+y>=P?x+y-P:x+y;}
int dec(int x,int y)  {return x-y< 0?x-y+P:x-y;}
int mul(int x,int y)  {return 1ll*x*y%P;}
int power(int a,int b,int ans=1){
	for(;b;b>>=1,a=mul(a,a))
		if(b&1)  ans=mul(ans,a);
	return ans;
}
int *w[22],C=21;
void prework(){
	for(int i=1;i<=C;++i)
		w[i]=new int[1<<(i-1)];
	int now=power(g,(P-1)/(1<<C));
	w[C][0]=1;
	for(int i=1;i<(1<<(C-1));++i)  w[C][i]=mul(w[C][i-1],now);
	for(int i=C-1;i;--i)
		for(int j=0;j<(1<<(i-1));++j)
			w[i][j]=w[i+1][j<<1];
}
void init(int lim){
	for(int i=0;i<lim;++i)
		pos[i]=(pos[i>>1]>>1)|((i&1)*(lim>>1));
}
void NTT(poly &f,int lim,int type){
	for(int i=0;i<lim;++i)
		if(pos[i]>i)  swap(f[i],f[pos[i]]);
	for(int mid=1,l=1;mid<lim;mid<<=1,++l){
		for(int i=0;i<lim;i+=(mid<<1)){
			for(int j=0;j<mid;++j){
				int p0=f[i+j],p1=mul(f[i+j+mid],w[l][j]);
				f[i+j]=add(p0,p1),f[i+j+mid]=dec(p0,p1);
			}
		}
	}
	if(type==-1&&(reverse(f.begin()+1,f.begin()+lim),1)){
		int inv=power(lim,P-2);
		for(int i=0;i<lim;++i)  f[i]=mul(f[i],inv);
	}
}
poly operator*(poly A,poly B){
	int len=A.size()+B.size()-2,lim=1;
	while(lim<=len)  lim<<=1;init(lim);
	A.resize(lim),NTT(A,lim,1);
	B.resize(lim),NTT(B,lim,1);
	for(int i=0;i<lim;++i)  A[i]=mul(A[i],B[i]);
	NTT(A,lim,-1),A.resize(len+1);
	return A;
}
poly Inv(poly A,int len){
	poly C,B(1,power(A[0],P-2));
	for(int lim=4;lim<(len<<2);lim<<=1){
		init(lim);
		C=A,C.resize(lim>>1);
		C.resize(lim),NTT(C,lim,1);
		B.resize(lim),NTT(B,lim,1);
		for(int i=0;i<lim;++i)  B[i]=mul(B[i],dec(2,mul(B[i],C[i])));
		NTT(B,lim,-1),B.resize(lim>>1);
	}
	B.resize(len);return B;
}
poly Deriv(poly A){
	for(int i=0;i<A.size()-1;++i)  A[i]=mul(A[i+1],i+1);
	A.pop_back();return A;
}
poly Integ(poly A){
	A.push_back(0);
	for(int i=A.size()-1;i;--i)  A[i]=mul(A[i-1],inv[i]);
	A[0]=0;return A;
}
poly Ln(poly A,int len){
	A=Integ(Deriv(A)*Inv(A,len)),A.resize(len);
	return A;
}
poly Exp(poly A,int len){
	poly B(1,1),C;A.resize(len<<1);
	for(int lim=2;lim<(len<<1);lim<<=1){
		C=Ln(B,lim);
		for(int i=0;i<lim;++i)  C[i]=dec(A[i],C[i]);
		C[0]=add(C[0],1),B=B*C;
		B.resize(lim);
	}
	B.resize(len);return B;
}
int main(){
	poly F(1,0);
	scanf("%d",&n),prework();
	fac[0]=fac[1]=inv[1]=1;
	for(int i=2;i<N;++i)  fac[i]=mul(fac[i-1],i);
	for(int i=2;i<N;++i)  inv[i]=mul(P-P/i,inv[P%i]);
	F.resize(n+1),F[1]=1;
	for(int i=2;i<=n;++i)  F[i]=inv[2];
	for(int i=4;i<=n;++i)  if(i%3!=0)  F[i]=add(F[i],inv[i<<1]);
	F=Exp(F,n+1);
	printf("%d",mul(F[n],fac[n]));
	return 0;
}