poj1655(Balancing Act + 树形dfs)

     题目链接：http://poj.org/problem?id=1655

     题意：有一颗N个节点的树，每个节点的Balancing是删除该节点后所有子树中节点数目最大的子树节点数目。问Balancing的最小值与该节点的编号。如果Balancing相同，输出节点编号较小的。

代码：

#include<stdio.h>
#include<string.h>
#define MAXN 20005
struct Edge{
	int v;
	int next;
}edge[MAXN*2];
int sum[MAXN*2],son[MAXN];
/*
sum[]记载除去"父节点的子树"的节点数量总和
son[]记录除去"父节点的子树"的所有子树中最大节点数
父节点在该题理解为DFS过程中的上一层节点
*/
int head[MAXN],visited[MAXN];
int tot;
int Max(int a,int b)
{
	return a>b ? a : b;
}
void addedge(int x,int y)
{
   tot++;
   edge[tot].v = y;
   edge[tot].next = head[x];
   head[x] = tot;
    
   tot++;
   edge[tot].v = x;
   edge[tot].next = head[y];
   head[y] = tot;
  
}
void dfs(int x)
{
   visited[x] = 1;
   int t;
   for(t=head[x];t!=-1;t=edge[t].next)
   {
	   if(!visited[edge[t].v])
	   {
           dfs(edge[t].v);
		   sum[x] += sum[edge[t].v];
           if(son[x] < sum[edge[t].v])
			   son[x] = sum[edge[t].v];
	   }
   }
}
int main()
{
	
	int ncase;
	scanf("%d",&ncase);
	while(ncase--)
	{
		int i,N;
		int x,y;
		int id,ans;
		scanf("%d",&N);
		tot = 0;
		memset(head,-1,sizeof(head));
		memset(edge,0,sizeof(edge));
		for(i=1;i<N;i++)
		{
			scanf("%d%d",&x,&y);
			addedge(x,y);
		}
		memset(visited,0,sizeof(visited));
		memset(son,0,sizeof(son));
		for(i=0;i<=N;i++)
			sum[i] = 1;
        dfs(1);
		id = 1;
		ans = son[1];
		for(i=2;i<N;i++)
			if(ans > Max(son[i],sum[1]-sum[i]))//sum[1]为整棵树的节点数量，sum[1]-sum[i]就是父节点子树的节点数目
			{
				ans = Max(son[i],sum[1]-sum[i]);
				id = i;
			}
		printf("%d %d\n",id,ans);
	}
	return 0;
}