浅谈树的重心----附核心代码----（c++）

前置知识：dfs

介绍一下树：

树是一种数据结构，它是由n(n≥0)个有限节点组成一个具有层次关系的集合。

树的重心：

定义：树的重心也叫树的质心。找到一个点,其所有的子树中最大的子树节点数最少,那么这个点就是这棵树的重心,删去重心后，生成的多棵树尽可能平衡。

看到定义后，相信各位都有自己的想法。事实上，最容易想到的莫过于用n个dfs暴力求出每一个点作为根其子树节点数的最大值并记录在数组中，之后通过比大小便可以找到树的重心。这种算法无疑是正确的，但明显，与之伴随的是高昂的时间代价（）。我们需要一种更优秀的算法。

大家思考一个问题，可不可以只用一遍dfs便解决问题呢？

显然是可以的。

首先定义一个d[]数组，d[u]记录节点u子树的节点个数；另有一个minn[]数组，minn[u]记录节点u的子树节点个数的最大值。

我们可以任意选一个节点作为根，从ta开始dfs，每一层都将d[当前节点编号]=1，搜到最底层后回溯，在回溯是可以用其儿子节点的d，maxn更新当前节点的的，maxn(具体实现见下面代码）。

但智慧的大佬们会发现，本人的思路似乎少了除其子树外的其他节点……是的！

其实剩下的节点数量，即u的“上方子树”的节点数量，正好是n（节点总数）-d[u]！

这样，我们便成功得到了d数组与maxn数组，之后仅需要简单的比大小即可。

上代码（题目为多组数据，每组数据一棵树，求树的重心
以及重心删除后得到的最大子树的节点个数）：

#include<bits/stdc++.h>
using namespace std;
const int N=50010;
int t,n,tot=0,head[N],ind,ans;
struct edge{
	int nxt,to;
}e[N*2];
void add(int u,int v){
	e[++tot]={head[u],v};
	head[u]=tot;
}
int d[N];
void dfs(int u,int fa){
	d[u]=1;
	int res=0;//Ñ¡Ôñ¿ªÒ»¸ö±äÁ¿Î¬»¤ 
	for(int i=head[u];i!=-1;i=e[i].nxt){
		int v=e[i].to;
		if(v==fa)continue;
		dfs(v,u);
		d[u]+=d[v];
		res=max(res,d[v]);
	}
	res=max(res,n-d[u]);
	if(res<ans||(res==ans&&u<ind)){
		ans=res;
		ind=u;
	}
} 
int main(){
	scanf("%d",&t);
	while(t--){
		tot=0;
		memset(head,-1,sizeof(head));
		memset(d,0,sizeof(d));
		ans=INT_MAX;
		scanf("%d",&n);
		for(int i=1;i<n;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			add(u,v);
			add(v,u);
		}
		dfs(1,0);
		printf("%d %d\n",ind,ans);
	}
	return 0;
}

谢谢大家！