poj2378(Tree Cutting)

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

      题意：一颗有N个节点的树，删除其中某个节点，如果其子树中节点数目最多的子树的节点数小于等于N/2，则该节点为合法节点，问该树中有多少个合法节点。

 代码：

#include<stdio.h>
#include<string.h>
#define MAXN 10005
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 i,N;
	while(scanf("%d",&N)!=EOF)
	{
		
		int x,y;
		int ans;
		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);
		int flag = 0;
		for(i=1;i<=N;i++)
		{
			ans = Max(son[i],sum[1]-sum[i]);//sum[1]为整棵树的节点数量，sum[1]-sum[i]就是父节点子树的节点数目
				if(ans <= N/2)
				{
					flag = 1;
					printf("%d\n",i);
				}
		}
		if(flag == 0)
			printf("NONE\n");
	}
	return 0;
}