[P3233][HNOI2014]世界树（虚树）

看到∑mi​≤300000自然联想到虚树，简单思考一下似乎可行，但剩下的部分貌似就比较麻烦。

首先对重新构建的虚树，考虑每个点应该被谁管。因为一个点既可以被它子树中的点管，也可以被子树以外的点管，所以我们做两次dfs。第一次先遍历子节点，再更新，记录的是一个点子树中最近的管他的点是谁。第二次先用从父节点得到的数据更新，再遍历子节点，是把子树之外可以管他的点也考虑进去。在第二次dfs时我们可以顺便统计一部分贡献，就是对于节点i，如果它其中一个子节点的子树不在虚树内，那这棵子树中一定没有可以管别人的点，那么显然，这棵子树中所有点都和节点i被同一个点管。

然后还有一部分没有统计，就是虚树的一条边上实际点，以及他们那些不在虚树上的子树。那么我们再dfs一次，如果一条边的两个节点被同一个点管，那没有问题；如果不是，我这个程序默认中间的所有点都和深度小的节点被同一个点管，我们需要修正一下。我们可以用倍增找到中间的分界点，然后重新统计这一段的贡献。

大概就是这样啦~

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=300010;
struct edge{
	int y,next;
}data[N<<1],data1[N];
struct query{
	int x,id;
}a1[N];
bool flag[N];
int n,q,m,num,num1,num2,h[N],h1[N],ans[N],dep[N],dfn[N],siz[N],fa[N][20],sta[N],top,dis[N],c1[N];
inline int read(){
	int x=0;char ch=getchar();
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
	return x;
}
inline void addedge(int u,int v){
	data[++num].y=v,data[num].next=h[u],h[u]=num;
	data[++num].y=u,data[num].next=h[v],h[v]=num;
}
inline void addedge1(int u,int v){
	data1[++num2].y=v,data1[num2].next=h1[u],h1[u]=num2;
}
inline bool cmp1(query a,query b){return dfn[a.x]<dfn[b.x];}
inline bool cmp2(query a,query b){return a.id<b.id;}
inline int lca(int u,int v){
	if(dep[u]<dep[v])swap(u,v);
	for(int i=19;i>=0;--i)if(dep[fa[u][i]]>=dep[v])u=fa[u][i];
	if(u==v)return u;
	for(int i=19;i>=0;--i)if(fa[u][i]!=fa[v][i])u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}
inline int getx(int u,int x){
	int y=0;
	for(int i=19;i>=0;--i)if(y+(1<<i)<=x)y+=1<<i,u=fa[u][i];
	return u;
}
void dfs1(int u,int d){
	dep[u]=d,dfn[u]=++num1,siz[u]=1;
	for(int i=h[u],v;i!=-1;i=data[i].next){
		v=data[i].y;
		if(fa[u][0]!=v){fa[v][0]=u;dfs1(v,d+1);siz[u]+=siz[v];}
	}
}
void dfs2(int u){
	if(flag[u])c1[u]=u,dis[u]=0;
	else c1[u]=0,dis[u]=N;
	for(int i=h1[u],v;i!=-1;i=data1[i].next){
		v=data1[i].y;dfs2(v);
		if(dis[u]>dis[v]+dep[v]-dep[u]||(dis[u]==dis[v]+dep[v]-dep[u]&&c1[u]>c1[v]))
			dis[u]=dis[v]+dep[v]-dep[u],c1[u]=c1[v];
	}
}
void dfs3(int u,int t,int d){
	int res=siz[u];
	if(d<dis[u]||(dis[u]==d&&t<c1[u]))dis[u]=d,c1[u]=t;
	else t=c1[u],d=dis[u];
	for(int i=h1[u],v;i!=-1;i=data1[i].next){
		v=data1[i].y;res-=siz[v];dfs3(v,t,d+dep[v]-dep[u]);
	}
	ans[c1[u]]+=res;
}
void dfs4(int u){
	for(int i=h1[u],v,x,xmid;i!=-1;i=data1[i].next){
		v=data1[i].y;
		if(c1[u]!=c1[v]){
			x=dep[v]-dep[u]+dis[u]+dis[v];
			if(x&1){
				x=x>>1,xmid=getx(c1[v],x);
				ans[c1[v]]+=siz[xmid]-siz[v],ans[c1[u]]-=siz[xmid]-siz[v];
			}else{
				x=x>>1;if(c1[u]<c1[v])--x;xmid=getx(c1[v],x);
				ans[c1[v]]+=siz[xmid]-siz[v],ans[c1[u]]-=siz[xmid]-siz[v];
			}
		}
		dfs4(v);
	}
	h1[u]=-1;
}
int main(){
	n=read();num=num1=0;memset(h,-1,sizeof h);
	for(int i=1,u,v;i<n;++i)u=read(),v=read(),addedge(u,v);
	fa[1][0]=0,dep[0]=-N;dfs1(1,1);
	for(int i=1;i<=19;++i)for(int j=1;j<=n;++j)fa[j][i]=fa[fa[j][i-1]][i-1];
	q=read();sta[1]=1;
	memset(h1,-1,sizeof h1);
	for(int i=1;i<=q;++i){
		m=read();top=1,num2=0;
		for(int j=1;j<=m;++j)a1[j].x=read(),a1[j].id=j,flag[a1[j].x]=true;
		sort(a1+1,a1+m+1,cmp1);
		for(int j=1,l;j<=m;++j){
			if(a1[j].x==1)continue;
			l=lca(a1[j].x,sta[top]);
			if(l!=sta[top]){
				while(top>1&&dfn[sta[top-1]]>=dfn[l])addedge1(sta[top-1],sta[top]),--top;
				if(l!=sta[top])addedge1(l,sta[top]),sta[top]=l;
			}
			sta[++top]=a1[j].x;
		}
		while(top>1)addedge1(sta[top-1],sta[top]),--top;
		dfs2(1);dfs3(1,0,N);dfs4(1);
		sort(a1+1,a1+m+1,cmp2);
		for(int j=1;j<=m;++j){printf("%d ",ans[a1[j].x]);flag[a1[j].x]=false,ans[a1[j].x]=0;}
		printf("\n");
	}
	return 0;
}