【bzoj1131】 [POI2008]Sta

Description

给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大

Input

给出一个数字N,代表有N个点.N<=1000000 下面N-1条边.

Output

输出你所找到的点,如果具有多个解,请输出编号最小的那个.

Sample Input

8
1 4
5 6
4 5
6 7
6 8
2 4
3 4

Sample Output

7

Solve

树形DP，可以O(1)从父亲转移到儿子

#include<iostream>
#include<cstdlib>
#include<cstdio>
using namespace std;
const int N=1000005;
int fi[N],ne[N<<1],j[N<<1],son[N],deep[N];
long long f[N];
int n,m,ans=0;
void dfs1(int ro,int fa){
    son[ro]=1;
    for (int u=fi[ro];u;u=ne[u]){
        if (j[u]==fa)continue;
        deep[j[u]]=deep[ro]+1;
        dfs1(j[u],ro);
        son[ro]+=son[j[u]];
    }
}
void dfs2(int ro,int fa){
    for (int u=fi[ro];u;u=ne[u]){
        if (j[u]==fa)continue;
        f[j[u]]=f[ro]-son[j[u]]+n-son[j[u]];
        dfs2(j[u],ro);
    }
}
int main (){
    scanf ("%d",&n);
    for (int i=1;i<n;++i){
        scanf ("%d%d",&j[i+n],&j[i]);
        ne[i]=fi[j[i+n]];fi[j[i+n]]=i;
        ne[i+n]=fi[j[i]];fi[j[i]]=i+n;
    }
    dfs1(1,0);
    for (int i=1;i<=n;++i)
        f[1]+=deep[i];
    dfs2(1,0);
    for (int i=1;i<=n;++i)
        if (f[i]>f[ans])ans=i;
    printf ("%d",ans);
    return 0;
}