[CSP day2T3]树的重心

树的重心

题解

考场上看到标题树的重心！！！什么，完了，忘记怎么求树的重心了。

好了，回归正题。

这道题出题人给的部分分很有讲究的。

• 对于前5个点，是个的暴力，是给笔者这样忘记树的重心打法的人送分的。
• 对于第6-8个点，我们可以用的打法打出来，也就是一个模板。
• 对于一条链的情况，也是十分好解决的，对于一条点数为奇数的链，它的重心一定为它中间的那个点，而偶数的链，它的重心一定为它中间的两个点，这样我们可以就通过的时间复杂度求出来了。
• 对于完美二叉树，笔者又想吐槽出题人了，如此阴险。数据范围给定了，经过严密的计算发现，262143这个数字是，即该树为一棵深度为17的满二叉树，我们就可以通过的时间复杂度算出来了。

此题非正解给了很高的分，足足有75分。

接下来我们讲一下正解：

因为对于一棵二叉树的根，若它本身不为重心，则重心一定在其重儿子的子树上。因为该树深度较小，我们可以重儿子倍增的方法，先dfs一遍，更新出其重儿子与子树大小，再一次dfs进行换根，对于一条要删的边，让其一个端点作为根，分别求出重心即可。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 3000005
using namespace std;
typedef long long LL;
#define int LL
int t,n,f[MAXN],f1[MAXN];
int from[MAXN],to[MAXN];
int nxt[MAXN],ans;
int tot,head[MAXN];
int hs1[MAXN],hs2[MAXN],hs3[MAXN];
int ps[MAXN][20],sum[MAXN],sum2[MAXN];
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(s>='0'&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
    x*=f;
}
void addEdge(int u,int v)
{
	from[++tot]=u;to[tot]=v;
	nxt[tot]=head[u];head[u]=tot;
}
void dfs(int u,int fa)
{
	sum[u]=1;f[u]=fa;
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v==fa) continue;
		dfs(v,u);sum[u]+=sum[v];
		if(sum[v]>sum[hs1[u]]) hs2[u]=hs1[u],hs1[u]=v;
		else if(sum[v]>sum[hs2[u]]) hs2[u]=v;
	} 
	ps[u][0]=hs1[u];
	for(int i=1;i<18;i++) 
		ps[u][i]=ps[ps[u][i-1]][i-1];
}
int tann(int u,int num)
{
	if(max(sum2[hs3[u]],num-sum2[u])<=num/2LL)
		return u;
	return 0;
}
void dfs2(int u,int fa)
{
	for(int i=head[u];i;i=nxt[i])
	{
		int v=to[i];
		if(v==fa) continue;
		sum2[u]=sum[1]-sum[v];f1[v]=f1[u]=0;
		if(hs1[u]==v) hs3[u]=hs2[u];
		else hs3[u]=hs1[u];
		if(sum2[fa]>sum2[hs3[u]]) hs3[u]=fa;
		ps[u][0]=hs3[u];
		for(int j=1;j<18;j++) 
			ps[u][j]=ps[ps[u][j-1]][j-1];
		int ss=u;
		for(int j=17;j>=0;j--)
			if(sum2[u]-sum2[ps[ss][j]]<=sum2[u]/2LL)
				ss=ps[ss][j];
		ans+=tann(hs3[ss],sum2[u])+tann(ss,sum2[u])+tann(f1[ss],sum2[u]);
		ss=v;
		for(int j=17;j>=0;j--)
			if(sum2[v]-sum2[ps[ss][j]]<=sum2[v]/2LL)
				ss=ps[ss][j];
		ans+=tann(hs3[ss],sum2[v])+tann(ss,sum2[v])+tann(f1[ss],sum2[v]);
		f1[u]=v;dfs2(v,u);
	}
	hs3[u]=ps[u][0]=hs1[u];f1[u]=f[u];
	for(int j=1;j<18;j++)
		ps[u][j]=ps[ps[u][j-1]][j-1];
	sum2[u]=sum[u];
}
signed main()
{
    read(t);
    while(t--)
    {
    	read(n);
		for(int i=1;i<=n;i++)
		{
			hs1[i]=hs2[i]=hs3[i]=0;
			sum[i]=sum2[i]=0;
			f[i]=f1[i]=0;
			head[i]=0;
		}
    	for(int i=1;i<n;i++)
    	{
    		int u,v;read(u);read(v);
    		addEdge(u,v);
    		addEdge(v,u);
		}	
		dfs(1,0);
		for(int i=1;i<=n;i++)
		{
			sum2[i]=sum[i];
			hs3[i]=hs1[i];
			f1[i]=f[i];
		}
		ans=0;dfs2(1,0);
		printf("%lld\n",ans);
	}
    return 0;
}

谢谢！！！