树的重心

一、题目

点此看题

二、解法

重心可以考虑从重儿子方面入手。
设$d[u][i]$为从$u$向重儿子跳$2^i$步，如果我们从一棵树的根开始跳，那么这样的方式$\log$次就可跳到重心。
然后考虑断开边$(u,v)$，则$u,v$就成了两个子树的根，只要有了$d$数组，就可以$\log$找答案。
考虑换根，什么是换根呢？就是你假装它是根，我们不需要去改变树的形态，而只需要修改一些信息，已达到快速换根的目的。
还有很多细节，可以看代码中的注释理解吧，时间复杂度$O(n\log n)$。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int MAXN = 300005;
int read()
{
    int x=0,flag=1;char c;
    while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
    while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^'0'),c=getchar();
    return x*flag;
}
int T,n,tot,f[MAXN],siz[MAXN],dn[MAXN][20];
int fa[MAXN],sum[MAXN],s1[MAXN],s2[MAXN],s3[MAXN];
long long ans;//注意开long long 
struct edge
{
	int v,next;
}e[MAXN*2];
void dfs(int u,int par)
{
	fa[u]=par;
	siz[u]=1;
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==par) continue;
		dfs(v,u);
		siz[u]+=siz[v];//get size 
		if(siz[v]>siz[s1[u]])
			s2[u]=s1[u],s1[u]=v;
		else if(siz[v]>siz[s2[u]])
			s2[u]=v;//找重儿子和次重儿子 
	}
	dn[u][0]=s1[u];//倍增 
	for(int i=1;i<=19;i++)
		dn[u][i]=dn[dn[u][i-1]][i-1];
}
int pd(int x,int tot)//判断是否满足重心的条件 
{
	return x*(max(sum[s3[x]],tot-sum[x])<=tot/2);
}
void dfs2(int u,int par)
{
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==par) continue;
		//假装断开(u,v)，再换根 
		sum[u]=siz[1]-siz[v];fa[u]=0;fa[v]=0;//得到新形态下的信息 
		if(v==s1[u]) s3[u]=s2[u];
		else s3[u]=s1[u];
		if(sum[par]>sum[s3[u]]) s3[u]=par;
		//找重儿子，先判断是否原重儿子是v，再用par来更新 
		dn[u][0]=s3[u];//重新倍增 
		for(int j=1;j<=19;j++)
			dn[u][j]=dn[dn[u][j-1]][j-1];
		int b=u;//以u为根来跳 
		for(int j=19;j>=0;j--)
			if(dn[b][j] && sum[u]-sum[dn[b][j]]<=sum[u]/2) b=dn[b][j];
		ans+=pd(b,sum[u])+pd(fa[b],sum[u]);
		//在满足子树外的联通块不大于sum[u]/2的情况下让sum[dn[b][j]]最小，就最优，因为跳的越远越小 
		//可能为b或fa[u]，重心可能有2个 
		b=v;//以v为跟来跳 
		for(int j=19;j>=0;j--)
			if(dn[b][j] && sum[v]-sum[dn[b][j]]<=sum[v]/2) b=dn[b][j];
		ans+=pd(b,sum[v])+pd(fa[b],sum[v]);
		fa[u]=v;//下一次断开的边就是(v,..)，所以v成为了u的新父亲 
		dfs2(v,u);
	}
	//回溯，还原所有信息 
	s3[u]=s1[u];sum[u]=siz[u];
	dn[u][0]=s1[u];fa[u]=par;
	for(int i=1;i<=19;i++)
		dn[u][i]=dn[dn[u][i-1]][i-1];
}
int main()
{
	T=read();
	while(T--)
	{
		n=read();
		tot=ans=0;
		for(int i=1;i<=n;i++)
		{
			siz[i]=f[i]=s1[i]=s2[i]=s3[i]=0;
			for(int j=0;j<=19;j++)
				dn[i][j]=0;
		}
		for(int i=2;i<=n;i++)//前向星 
		{
			int u=read(),v=read();
			e[++tot]=edge{v,f[u]},f[u]=tot;
			e[++tot]=edge{u,f[v]},f[v]=tot;
		}
		dfs(1,0);
		for(int i=1;i<=n;i++)
			sum[i]=siz[i],s3[i]=s1[i];
		//sum是当前树形态的size,s3是当前树形态的重儿子 (fa同理) 
		dfs2(1,0);
		printf("%lld\n",ans);
	}
}