树的重心【poj1655】

定义

树的重心也叫树的质心。对于一棵树n个节点的无根树，找到一个点，使得把树变成以该点为根的有根树时，最大子树的结点数最小。换句话说，删除这个点后最大连通块（一定是树）的结点数最小。

数的重心求法其实非常简单，用到了简单的树形dp的思想。

你只要进行一次dfs，一边求以当前节点为根的子树的节点数，一边统计出子树中最大的有max{f[j]}个节点（j为i的儿子）和i节点上方的子树大小为n-f[i]个节点。这样就可以找到重心了。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,cnt,T,a,b,mind,min0;
int d[50035],head[50035],nxt[50035],to[50035];
void add(int x,int y)
{
	cnt++;nxt[cnt]=head[x];to[cnt]=y;head[x]=cnt;
}
void dfs(int u,int fa)
{
	d[u]=1;
	int max0=0;
	for (int i=head[u];i;i=nxt[i])
	{
		int son=to[i];
		if (son==fa) continue;
		dfs(son,u);
		d[u]=d[u]+d[son];//统计以u节点为根的子树的节点个数
		max0=max(max0,d[son]);
	}
	max0=max(max0,n-d[u]);
	if (min0>max0)
	{
		min0=max0;
		mind=u;
	}
}
int main()
{
	scanf("%d",&T);
	while (T>0)
	{
		T--;
		min0=1e9+7;
		cnt=0;
		memset(head,0,sizeof(head));
		scanf("%d",&n);
		for (int i=1;i<=n-1;i++) 
		{
			scanf("%d%d",&a,&b);
			add(a,b);
			add(b,a);
		}
		 dfs(1,0);
		printf("%d %d\n",mind,min0);
	}
}