6435. 【NOIP2019提高组正式赛day2】树的重心（centroid）

Description

Input

Output

输出到文件 centroid.out 中。
共 T 行，每行一个整数，第 i 行的整数表示：第 i 组数据给出的树单独删去每条边后，分裂出的两个子树的重心编号和之和。 

Sample Input

2
5 
1 2
2 3
2 4
3 5
7
1 2
1 3
1 4
3 5
3 6
6 7
 

Sample Output

32
56

Data Constraint

Solution

首先对于一棵树的重心一定在根节点的重链上，所以直接倍增跳重链，当上面的节点数小于等于total除以二就可以跳（这是重儿子的定义）。最后再判断一下跳到的点的父亲是否合法即可。

对于删除的边（记为x,y），深度大的点（记为y）可以直接按照普通找重心的方法找，而对于其余部分的重心可以从根节点同样开始跳重链，但保证不跳进删除的边深度较大的点所在的子树(y的子树内)，这个可以用dfs序实现，而当跳到了x这个点时，那么就跳它的次重儿子再继续跳重儿子即可。

 

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define I int
#define ll long long	
#define F(i,a,b) for(I i=a;i<=b;i++)
#define Fd(i,a,b) for(I i=a;i>=b;i--)
#define mem(a,b) memset(a,b,sizeof(a))
#define N 300010
using namespace std;
I T,n,x[N],y[N],t[N*2],nx[N*2],l[N],sz[N],f[N],cs[N],top[N],st[N],ed[N],cnt,tot;
I s[N][20];
ll ans;
void rd(I &x){
	x=0;char ch=getchar();
	while(ch<'0'||ch>'9') ch=getchar();
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
}
void add(I x,I y){t[++tot]=y;nx[tot]=l[x];l[x]=tot;}
void dg(I x,I y){
	s[x][0]=-1;sz[x]=1;st[x]=++cnt;I z;
	for(I k=l[x];k;k=nx[k]) if(t[k]!=y){
		f[z=t[k]]=x;
		dg(z,x);
		sz[x]+=sz[z];
		if(s[x][0]==-1||sz[z]>sz[s[x][0]]){
			if(s[x][0]!=-1) cs[x]=s[x][0];
			s[x][0]=z;
		}
		else if(sz[z]>sz[cs[x]]) cs[x]=z;
	}
	ed[x]=++cnt;
}
void work(I x){
	Fd(i,19,0) if(tot-sz[s[x][i]]<=tot/2) x=s[x][i];
	ans+=x+f[x]*(sz[x]==tot/2);
}
void jump(I x,I y){
	Fd(i,19,0){
		if(s[x][i]<1||st[s[x][i]]>=st[y]&&st[s[x][i]]<=ed[y]) continue;
		if(st[y]>st[s[x][i]]&&ed[y]<ed[s[x][i]]){if(n-sz[s[x][i]]<=tot/2) x=s[x][i];}
		else{if(tot-sz[s[x][i]]<=tot/2) x=s[x][i];}
	}
	if(tot-sz[cs[x]]<=tot/2){work(cs[x]);return;}
	if(st[y]>st[x]&&ed[y]<ed[x]){ans+=x*(n-sz[x]<=tot/2)+f[x]*(sz[x]-sz[y]<=tot/2);}
	else{ans+=x*(tot-sz[x]<=tot/2)+f[x]*(sz[x]<=tot/2);}
}
I main(){
	freopen("centroid.in","r",stdin);
	freopen("centroid.out","w",stdout);
	rd(T);
	while(T--){
		rd(n);
		tot=ans=cnt=0;
		mem(l,0);
		mem(nx,0);
		mem(s,0);//重儿子倍增数组
		mem(cs,0);//次重儿子 
		F(i,1,n-1){
			rd(x[i]),rd(y[i]);
			add(x[i],y[i]);add(y[i],x[i]);
		}
		dg(1,0);
		F(j,1,19){
			F(i,1,n) if(s[i][j-1]!=-1) s[i][j]=s[s[i][j-1]][j-1];	
		}
		//预处理 
		F(i,1,n-1){//枚举边 
			if(st[x[i]]>st[y[i]]) swap(x[i],y[i]);
			tot=sz[y[i]];
			work(y[i]);
			tot=n-sz[y[i]];
			jump(1,y[i]);
		}
		printf("%lld\n",ans);
	}
	return 0;
}

---

---

作者：zsjzliziyang 
QQ:1634151125 
转载及修改请注明 
本文地址:https://blog.csdn.net/zsjzliziyang/article/details/103321582