POJ3107 树的重心

看代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;
typedef long long LL;
const int maxN=5e4+5;
int N,dp[maxN],ans;
struct DM
{
    int to,next;
}path[maxN<<1];
int head[maxN],tol;
void add(int u,int v)
{
    path[tol]={v,head[u]};
    head[u]=tol++;
}
void init()
{
    memset(head,-1,sizeof(head));
    tol=0;
    ans=N+1;
}
int dfs(int u,int fa)
{
    int sum=1;
    for(int i=head[u];i!=-1;i=path[i].next)
    {
        int v=path[i].to;
        if(v==fa)continue;
        int d=dfs(v,u);
        sum+=d;
        dp[u]=max(dp[u],d);
    }
    dp[u]=max(dp[u],N-sum);
    ans=min(ans,dp[u]);
    return sum;
}
int main()
{
    scanf("%d",&N);
    init();
    int u,v;
    for(int i=1;i<N;i++)
    {
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(1,1);
    for(int i=1;i<=N;i++)if(dp[i]==ans)printf("%d ",i);
    return 0;
}