1131: [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

 

树形DP。。

#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<set>
#define inf 1000000000
#define maxn 1000000+5
#define maxm 2000000+5
#define eps 1e-10
#define ll long long
#define for0(i,n) for(int i=0;i<=(n);i++)
#define for1(i,n) for(int i=1;i<=(n);i++)
#define for2(i,x,y) for(int i=(x);i<=(y);i++)
#define for3(i,x,y) for(int i=(x);i>=(y);i--)
#define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go)
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{
    int go,next;
}e[maxm];
int n,tot,ans,head[maxn],size[maxn],fa[maxn];
ll sum,f[maxn],g[maxn];
void insert(int u,int v){
    e[++tot]=(edge){v,head[u]};head[u]=tot;
    e[++tot]=(edge){u,head[v]};head[v]=tot;
}
void dfs1(int x){
    f[x]=size[x]=1;
    for4(i,x)
        if(y!=fa[x]){
            fa[y]=x;
            dfs1(y);
            size[x]+=size[y];
            f[x]+=f[y]+size[y];
        }
}
void dfs2(int x){
    for4(i,x)
        if(y!=fa[x]){
            g[y]=f[x]+g[x]-f[y]-size[y]+n-size[y];
            dfs2(y);
        }
}
int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    n=read();
    for1(i,n-1){
        int u=read(),v=read();
        insert(u,v);
    }
    dfs1(1);
    dfs2(1);
    for1(i,n){
        g[i]+=f[i];
        if(g[i]>sum){
            sum=g[i];ans=i;
        }
    }
    printf("%d",ans);
    return 0;
}