poj2378（树形DP）

题目链接:题目传送门

题目大意：删除一个节点后，他的最大子树的结点个数小于等于一半。以升序的方式输出所有满足的结点。如果没有的话输出NONE

思路：即找树的重心的算法 搜索计算每个结点子树的大小

#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
#define maxn 50000+10
int n;
int head[maxn];
struct node
{
    int v,nxt;
}e[maxn*2];
int cnt;
void add(int u,int v)
{
    e[cnt].v=v;
    e[cnt].nxt=head[u];
    head[u]=cnt++;
}
bool vis[maxn];
int sz[maxn];
int ans,sum[maxn];
void dfs(int u)
{
    vis[u]=1;
    sz[u]=1;
    int t=-1;
    for(int i=head[u];i!=-1;i=e[i].nxt)
    {
        int v=e[i].v;
        if(vis[v])continue;
        dfs(v);
        sz[u]+=sz[v];
        t=max(t,sz[v]);
    }
    t=max(t,n-sz[u]);
    if(t<=n/2)
    {
        sum[ans++]=u;
    }
}
int main()
{
   while(~scanf("%d",&n))
   {
       memset(head,-1,sizeof head);
       cnt=0;
       memset(vis,0,sizeof vis);
       for(int i=1;i<n;i++)
       {
           int u,v;
           scanf("%d%d",&u,&v);
           add(u,v);
           add(v,u);
       }
       ans=0;
       dfs(1);
       sort(sum,sum+ans);
       if(ans==0)
       {
           printf("NONE\n");
       }
       for(int i=0;i<ans;i++)
       {
           printf("%d\n",sum[i]);
       }
   }
    return 0;
}