【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

题解
dp乱搞题
快读超时？？

Down 记录子树深度和
Up 记录父亲方向为子树的深度和

代码

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
inline 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=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int size[1000005],dep[1000005],n,u,v;
ll Down[1000005],Up[1000005],ans[1000005];
int tot,Head[1000005],Next[2000005],ret[2000005];
inline void ins(int u,int v){ret[++tot]=v;Next[tot]=Head[u];Head[u]=tot;}
void dfs(int u,int f)
{
    Down[u]=dep[u]=dep[f]+1;size[u]=1;
    for (int i=Head[u];i;i=Next[i])
    {
        if (ret[i]==f) continue;
        dfs(ret[i],u);
        size[u]+=size[ret[i]];
        Down[u]+=dep[ret[i]];
    }
}
void dp(int u,int f)
{
    ans[u]=Down[u]-(ll)dep[u]*size[u]+Up[u];
    for (int i=Head[u];i;i=Next[i])
    {
        if (ret[i]==f) continue;
        int v=ret[i];
        Up[v]=Down[u]-Down[v]-(ll)(dep[u]-1)*(size[u]-size[v])+Up[u]+(n-size[u]);
        dp(v,u);
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        ins(u,v);ins(v,u);
    }
    dep[0]=-1;
    dfs(1,0);
    dp(1,0);
    int ANS=0;ans[0]=-1;
    for (int i=1;i<=n;i++) if (ans[i]>ans[ANS]) ANS=i;
    cout<<ANS;
    return 0;
}