bzoj 1131 [POI2008]Sta 树形dp 转移根模板题

[POI2008]Sta

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 1889  Solved: 729
[Submit][Status][Discuss]

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

HINT

 

题解：这是一道裸题

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<algorithm>
 #include<cmath>
 #include<vector>

 #define N 1000007
 #define ll long long
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x*f;
 }

 int n,m,id;
 int cnt,hed[N],rea[N<<],nxt[N<<];
 ll sum[N],dep[N],siz[N];

 void add(int u,int v)
 {
     nxt[++cnt]=hed[u];
     hed[u]=cnt;
     rea[cnt]=v;
 }
 void add_two_way(int x,int y)
 {
     add(x,y);
     add(y,x);
 }
 void dfs1(int u,int fa)
 {
     siz[u]=;
     for (int i=hed[u];i!=-;i=nxt[i])
     {
         int v=rea[i];
         if (v==fa) continue;
         dep[v]=dep[u]+;
         dfs1(v,u);
         sum[u]+=sum[v],siz[u]+=siz[v];
     }
     sum[u]+=dep[u];
 }
 void dfs2(int u,int fa)
 {
     for (int i=hed[u];~i;i=nxt[i])
     {
         int v=rea[i];
         if (v==fa) continue;
         sum[v]=sum[u]-siz[v]+n-siz[v];
         dfs2(v,u);
     }
 }
 int main()
 {
     memset(hed,-,sizeof(hed));
     n=read();
     for (int i=;i<n;i++)
         add_two_way(read(),read());
     dfs1(,),dfs2(,);
     for (int i=;i<=n;i++) if (sum[i]>sum[id]) id=i;
     printf("%d\n",id);
 }