（树形dp，求重心）POJ3107Godfather

POJ3107Godfather

题意&思路：

就是求树的重心的模板题。重心就是一个结点的所有子树中最大子树结点最小，我们可以理解为删除这个结点后，能够得到的子树尽可能的平衡。
我们用sz[i]表示i结点的子树大小，可以的到s[i]=1+∑s[j],j为i的子节点。
所以我们可以假设1为根节点，dfs遍历整棵树，得到每个子结点的子树sz[i]。
然后求每个结点的最大子树，对于第u个结点及他的子节点v_i。我们可得mx=max(mx,sz[v_i])，mx初始值为n-sz[u]，可以理解为整棵树一共有n个结点，我们在刚开始是假设1为根节点统计的子节点个数，所以n-sz[u]就是以u为根节点，子结点中存在1的子树的大小。

代码：

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#include<ctype.h>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
const int N=1e5+10;
const int mod=1e7+9;
const int maxn=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
const int inf=99999999;
using namespace std;
struct edge
{
	int next,to;
}tree[N];
int head[N]={0},fa[N]={0},len=0,sz[N]={0};
void add(int u,int v)
{
	tree[++len]={head[u],v};
	head[u]=len;
}
void dfs(int u,int f)
{
	int i;
	sz[u]=1;
	fa[u]=f;
	for(i=head[u];i;i=tree[i].next)
	{
		int v=tree[i].to;
		if(v!=f)
		{
			dfs(v,u);
			sz[u]+=sz[v];
		}
	}
}
int main()
{
	int n,i,j;
	int res[N],l,mn=maxn;
	scanf("%d",&n);
	for(i=1;i<=n-1;i++)
	{
		int u,v,w;
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	dfs(1,0);
	for(i=1;i<=n;i++)
	{
		int mx=n-sz[i];
		for(j=head[i];j;j=tree[j].next)
		{
			int v=tree[j].to;
			if(v!=fa[i])
				mx=max(mx,sz[v]);
		}	
		if(mx<mn)
		{
			l=0;
			mn=mx;
			res[++l]=i;
		}
		else if(mx==mn)
			res[++l]=i;
	}
	sort(res+1,res+l+1);
	for(i=1;i<=l;i++)
		printf("%d ",res[i]);
	printf("\n");
	return 0;
}