poj 3170 Godfather(树形dp)

和poj1655一样

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
typedef __int64 lld;
#define oo 0x3f3f3f3f
#define maxn 50005
#define maxm 205
int dp[maxn];
struct EDGE
{
    int v,next;
}E[maxn<<1];
int sum[maxn];
int head[maxn],tol,n;

void inst()
{
    memset(head,-1,sizeof head);
    tol=0;
    memset(dp,0,sizeof dp);
}

void add_edge(int u,int v)
{
    E[tol].v=v;
    E[tol].next=head[u];
    head[u]=tol++;
}

int dfs(int u,int pre)
{
    sum[u]=1;
    for(int i=head[u];i!=-1;i=E[i].next)
    {
        int v=E[i].v;
        if(v==pre) continue;
        sum[u]+=dfs(v,u);
    }
    return sum[u];
}

void tree_dp(int u,int pre)
{
    for(int i=head[u];i!=-1;i=E[i].next)
    {
         int v=E[i].v;
         if(v!=pre)
         {
             dp[u]=max(dp[u],sum[v]);
             tree_dp(v,u);
         }
         else
            dp[u]=max(dp[u],n-sum[u]);
    }
}

int main()
{
    int u,v,ans,pos,f,T;

    while(scanf("%d",&n)!=EOF)
    {
        inst();
        for(int i=1;i<=n-1;i++)
        {
            scanf("%d%d",&u,&v);
            add_edge(u,v);
            add_edge(v,u);
        }
        dfs(1,-1);
        tree_dp(1,-1);
        ans=oo;
        f=1;
        for(int i=1;i<=n;i++)
            ans=min(dp[i],ans);
        for(int i=1;i<=n;i++)
        {
            if(dp[i]==ans)
            {
                if(f) f=0;
                else printf(" ");
                printf("%d",i);
            }
        }
        puts("");
    }
    return 0;
}