POI 2008 STA （树形DP入门）

 

#include <bits/stdc++.h>

using namespace std;

#define inf 0x3f3f3f3f

const int maxn=1e6+5;

int n,cnt;
int head[maxn];

struct node{
    int next;
    int to;
}p[maxn*2];

void addedge(int a,int b){
    p[++cnt].to=b;
    p[cnt].next=head[a];
    head[a]=cnt;
}

long long f[maxn];
int size[maxn];

inline void init(){
    cnt=0;
    memset(head,-1,sizeof(head));
}

void dfs1(int u,int fa){
    f[u]=size[u]=1;
    for(int e=head[u];e!=-1;e=p[e].next){
        if(p[e].to!=fa){
            dfs1(p[e].to,u);
            size[u]+=size[p[e].to];
            f[u]+=f[p[e].to]+size[p[e].to];
        }
    }
}

void dfs2(int u,int fa){
    for(int e=head[u];e!=-1;e=p[e].next){
        if(p[e].to!=fa){
            f[p[e].to]=f[u]+n-2*size[p[e].to];
            dfs2(p[e].to,u);
        }
    }
}

int main()
{
    init();
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        addedge(a,b);
        addedge(b,a);
    }
    dfs1(1,0);
    dfs2(1,0);
    int ans=0;
    f[0]=-1;
    for(int i=1;i<=n;i++){
        if(f[ans]<f[i]) ans=i;
    }
    printf("%d",ans);
    return 0;
}
/*
*/