Forest Game_gameforest模板-CSDN博客

本文链接：https://blog.csdn.net/Tan_tan_tann/article/details/113944317

Forest Game

题解

首先，由于每个点会被选当且仅当它到根上的所有点都没被选。
它到根上的点总共有 $dep_{i}$ 个，故它给期望的贡献为 $\frac{1}{dep_{i}}$ 。
这棵树又是无根树，所以答案为 $\sum_{i=1}^{n}\sum_{j=1}^{n}\frac{1}{1+dis(i,j)}$ 。

考虑如何求树上所有点之间的距离。
这不是点分治板子题吗？
直接一个点分治套上去，我们又发现对于重心统计路径长度时还需要合并两条链，对于菊花图就是 $O(n^2)$ 的了。
考虑FFT，两条链的合并不就相当于多项式乘法吗。
注意每次FFT时都要重新计算一下 $l i m$ ，不然如果一直用起点的 $l i m$ 的话每次都是 $O\left(n^2\right)$ 的了，点分治白打了。

总时间复杂度 $O\left(nlog^2n\right)$ 。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
#define MAXN 200010
#define MAXM 800010
#define lowbit(x) (x&-x)
#define reg register
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int,int> pii;
const int INF=0x7f7f7f7f;
const int mo=1e9+7;
const double PI=acos(-1.0);
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int n,head[MAXN],tot,siz[MAXN],root;LL sum[MAXN],lim;
int sta[MAXN],stak,S,mxson[MAXN],mx,dis[MAXN],ans,rev[MAXM];
bool vis[MAXN],vp[MAXN];
queue<int> q;
inline int add(const int x,const int y,const int p){return x+y<p?x+y:x+y-p;}
inline int qkpow(int a,int s,const int p){int t=1;while(s){if(s&1)t=1ll*a*t%p;a=1ll*a*a%p;s>>=1;}return t;}
struct edge{int to,nxt;}e[MAXN<<1];
void addEdge(int u,int v){e[++tot]=(edge){v,head[u]};head[u]=tot;}
void getRoot(int u,int fa){
	siz[u]=1;mxson[u]=0;
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa||vis[v])continue;getRoot(v,u);
		siz[u]+=siz[v];mxson[u]=max(siz[v],mxson[u]);
	}
	mxson[u]=max(mxson[u],S-siz[u]);
	if(mxson[u]<mx){mx=mxson[u];root=u;}
}
struct cp{
	double x,y;cp(double X=0,double Y=0){x=X;y=Y;}
	inline friend cp operator + (const cp &a,const cp &b){return cp(a.x+b.x,a.y+b.y);}
	inline friend cp operator - (const cp &a,const cp &b){return cp(a.x-b.x,a.y-b.y);}
	inline friend cp operator * (const cp &a,const cp &b){return cp(a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x);}
}F[MAXM];
inline void FFT(cp *A,int typ){
	for(int i=0;i<lim;i++)if(i<rev[i])swap(A[i],A[rev[i]]);
	for(int mid=1;mid<lim;mid<<=1){
		const cp Wn(cos(PI/mid),typ*sin(PI/mid));
		for(int R=mid<<1,j=0;j<lim;j+=R){
			cp w(1,0);
			for(int k=0;k<mid;k++,w=w*Wn){
				cp x=A[j+k],y=w*A[j+mid+k];
				A[j+k]=x+y;A[j+mid+k]=x-y;
			}
		}
	}
}
void sakura(int st,int w,int ds){
	while(!q.empty())q.pop();q.push(st);vp[st]=1;sta[stak=1]=st;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=head[u];i;i=e[i].nxt){
			int v=e[i].to;if(vis[v]||vp[v])continue;
			dis[v]=dis[u]+1;vp[v]=1;sta[++stak]=v;q.push(v);
		}
	}
	int maxx=0;for(int i=1;i<=stak;i++)maxx=max(maxx,dis[sta[i]]+ds);
	int L=0;lim=1;while(lim<=2*maxx+1)lim<<=1,L++;
	rev[0]=0;for(int i=1;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<L-1);
	for(int i=1;i<=stak;i++)F[dis[sta[i]]+ds].x++,vp[sta[i]]=dis[sta[i]]=0;
	FFT(F,1);for(int i=0;i<lim;i++)F[i]=F[i]*F[i];FFT(F,-1);
	for(int i=0;i<lim;i++)sum[i]+=1ll*w*((LL)(F[i].x/lim+0.5));
	for(int i=0;i<lim;i++)F[i]=cp(0,0);
}
void devide(int u){
	vis[u]=1;sakura(u,1,0);
	for(int i=head[u];i;i=e[i].nxt)
		if(!vis[e[i].to])sakura(e[i].to,-1,1);
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(vis[v])continue;
		S=siz[v];mx=INF;getRoot(v,0);devide(root);
	}
}
signed main(){
	read(n);for(int i=1,u,v;i<n;i++)read(u),read(v),addEdge(u,v),addEdge(v,u);
	S=n;mx=INF;getRoot(1,0);devide(root);
	for(int i=0;i<n;i++)ans=add(ans,1ll*sum[i]%mo*qkpow(i+1,mo-2,mo)%mo,mo);
	for(int i=1;i<=n;i++)ans=1ll*ans*i%mo;printf("%d\n",ans);
	return 0;
}