题目链接: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;
}