[HNOI2014]世界树 (虚树DP+倍增)

世界树

世界树是一棵无比巨大的树，它伸出的枝干构成了整个世界。在这里，生存着各种各样的种族和生灵，他们共同信奉着绝对公正公平的女神艾莉森，在他们的信条里，公平是使世界树能够生生不息、持续运转的根本基石。
世界树的形态可以用一个数学模型来描述：世界树中有n个种族，种族的编号分别从1到n，分别生活在编号为1到n的聚居地上，种族的编号与其聚居地的编号相同。有的聚居地之间有双向的道路相连，道路的长度为1。保证连接的方式会形成一棵树结构，即所有的聚居地之间可以互相到达，并且不会出现环。定义两个聚居地之间的距离为连接他们的道路的长度；例如，若聚居地a和b之间有道路，b和c之间有道路，因为每条道路长度为1而且又不可能出现环，所卧a与c之间的距离为2。
出于对公平的考虑，第i年，世界树的国王需要授权m[i]个种族的聚居地为临时议事处。对于某个种族x（x为种族的编号），如果距离该种族最近的临时议事处为y（y为议事处所在聚居地的编号），则种族x将接受y议事处的管辖（如果有多个临时议事处到该聚居地的距离一样，则y为其中编号最小的临时议事处）。
现在国王想知道，在q年的时间里，每一年完成授权后，当年每个临时议事处将会管理多少个种族（议事处所在的聚居地也将接受该议事处管理）。 现在这个任务交给了以智慧著称的灵长类的你：程序猿。请帮国王完成这个任务吧。

Input

第一行为一个正整数n，表示世界树中种族的个数。
接下来n-l行，每行两个正整数x，y，表示x聚居地与y聚居地之间有一条长度为1的双
向道路。接下来一行为一个正整数q，表示国王询问的年数。
接下来q块，每块两行：
第i块的第一行为1个正整数m[i]，表示第i年授权的临时议事处的个数。
第i块的第二行为m[i]个正整数h[l]、h[2]、…、h[m[i]]，表示被授权为临时议事处的聚居地编号（保证互不相同）。

Output

输出包含q行，第i行为m[i]个整数，该行的第j(j=1，2…，，m[i])个数表示第i年被授权的聚居地h[j]的临时议事处管理的种族个数。

Sample Input

10 
2 1 
3 2 
4 3 
5 4 
6 1 
7 3 
8 3 
9 4 
10 1 
5 
2 
6 1   
5 
2 7 3 6 9   
1 
8  
4 
8 7 10 3   
5 
2 9 3 5 8 

Sample Output

1 9   
3 1 4 1 1   
10  
1 1 3 5   
4 1 3 1 1 

Hint

N<=300000, q<=300000,m[1]+m[2]+…+m[q]<=300000

 

 

 

题解

思路很简单，直接建虚树

先把每个点的最近管辖点找到（换根DP一下）

然后来考虑每条边是否被分裂为两部分

即判断它的两个端点的最近管辖点是不是同一个点

如果是，就直接把这条边的答案加到那个管辖点上

如果不是，就需要倍增来计算这条边断开的位置，分别把答案加到两个管辖点

 

代码（大码量选手。。。）：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
inline int gi()
{
	char c;int num=0,flg=1;
	while((c=getchar())<'0'||c>'9')if(c=='-')flg=-1;
	while(c>='0'&&c<='9'){num=num*10+c-48;c=getchar();}
	return num*flg;
}
#define N 300005
#define LOG 19
#define INF 0x3f3f3f3f
int fir[N],to[2*N],nxt[2*N],cnt;
int dep[N],siz[N],f[N][LOG+2];
int dfn[N],dc;
int stk[N],TP,del[N],dcon;bool vis[N];

int tmp[N],bel[N],mi[N],g[N],rem[N];

void adde(int a,int b)
{
	to[++cnt]=b;nxt[cnt]=fir[a];fir[a]=cnt;
	//printf("%d %d\n",a,b);
}
void dfs1(int u)
{
	dep[u]=dep[f[u][0]]+1;
	siz[u]=1;
	dfn[u]=++dc;
	for(int v,p=fir[u];p;p=nxt[p]){
		v=to[p];
		if(v!=f[u][0]){
			f[v][0]=u;
			dfs1(v);
			siz[u]+=siz[v];
		}
	}
}
int getk(int x,int k)
{
	for(int i=LOG;i>=0;i--)
		if(k&(1<<i)) x=f[x][i];
	return x;
}
int getd(int x,int d)
{
	return getk(x,dep[x]-d);
}
int LCA(int x,int y)
{
	if(dep[x]>dep[y])
		x=getd(x,dep[y]);
	else if(dep[x]<dep[y])
		y=getd(y,dep[x]);
	if(x==y) return x;
	for(int i=LOG;i>=0;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
int dis(int x,int y)
{
	return dep[x]+dep[y]-2*dep[LCA(x,y)];
}
bool cmp(int x,int y){return dfn[x]<dfn[y];}
void build()
{
	sort(del+1,del+dcon+1,cmp);
	TP=0;
	int i,lca,tmp=dcon;
	if(del[1]!=1)del[++dcon]=1,stk[++TP]=1;
	for(i=1;i<=tmp;i++){
		vis[del[i]]=1;
		if(TP<=1||(lca=LCA(stk[TP],del[i]))==stk[TP]){
			stk[++TP]=del[i];
			continue;
		}
		while(TP>1&&dep[stk[TP-1]]>=dep[lca]){
			adde(stk[TP-1],stk[TP]);
			TP--;
		}
		if(lca!=stk[TP]){
			adde(lca,stk[TP]);
			stk[TP]=lca;del[++dcon]=lca;
		}
		stk[++TP]=del[i];
	}
	while(TP>1){
		adde(stk[TP-1],stk[TP]);
		TP--;
	}
}
void clean()
{
	for(int i=1;i<=dcon;i++){
		g[del[i]]=bel[del[i]]=fir[del[i]]=vis[del[i]]=0;
		rem[del[i]]=0;
		mi[del[i]]=INF;
	}
	dcon=0;cnt=0;
}

void dfs2(int u)
{
	if(vis[u])bel[u]=u,mi[u]=0;
	for(int v,p=fir[u];p;p=nxt[p]){
		v=to[p];
		dfs2(v);
		int d1=dep[v]-dep[u]+mi[v];
		if(mi[u]>d1||(mi[u]==d1&&bel[u]>bel[v])){
			mi[u]=d1;
			bel[u]=bel[v];
		}
	}
}
void dfs3(int u)
{
	for(int v,p=fir[u];p;p=nxt[p]){
		v=to[p];
		int d1=mi[u]+dep[v]-dep[u];
		if(mi[v]>d1||(mi[v]==d1&&bel[v]>bel[u])){
			mi[v]=d1;
			bel[v]=bel[u];
		}
		dfs3(v);
	}
}
void solve(int u)
{
	rem[u]=siz[u];
	for(int p=fir[u];p;p=nxt[p]){
		solve(to[p]);
		int x=to[p],mid=to[p];
		int lp=bel[to[p]],rp=bel[u];
		
		x=getd(x,dep[u]+1);
		rem[u]-=siz[x];
		if(lp==rp){
			g[lp]+=siz[x]-siz[to[p]];
			continue;
		}
		for(int i=LOG;i>=0;i--){
			if(dep[f[mid][i]]<dep[u])continue;
			if(dis(f[mid][i],rp)>dis(f[mid][i],lp)||(dis(f[mid][i],rp)==dis(f[mid][i],lp)&&rp>lp))
				mid=f[mid][i];
		}
		g[lp]+=siz[mid]-siz[to[p]];
		g[rp]+=siz[x]-siz[mid];
	}
	g[bel[u]]+=rem[u];
}
int main()
{
	int n,Q,m,i,j,u,v;
	n=gi();
	for(i=1;i<n;i++){
		u=gi();v=gi();
		adde(u,v);adde(v,u);
	}
	dfs1(1);
	for(j=1;j<=LOG;j++)
		for(i=1;i<=n;i++)
			f[i][j]=f[f[i][j-1]][j-1];
	
	memset(fir,0,sizeof(fir));cnt=0;
	memset(mi,0x3f,sizeof(mi));
	Q=gi();
	while(Q--){
		m=gi();
		dcon=0;
		for(i=1;i<=m;i++){
			tmp[i]=gi();
			del[++dcon]=tmp[i];
		}
		build();
		dfs2(1);
		dfs3(1);
		solve(1);
		printf("%d",g[tmp[1]]);
		for(i=2;i<=m;i++)
			printf(" %d",g[tmp[i]]);
		printf("\n");
		clean();
	}
}