【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;}