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=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 n,m,id;
int cnt,hed[N],rea[N<<1],nxt[N<<1];
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]=1;
    for (int i=hed[u];i!=-1;i=nxt[i])
    {
        int v=rea[i];
        if (v==fa) continue;
        dep[v]=dep[u]+1;
        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,-1,sizeof(hed));
    n=read();
    for (int i=1;i<n;i++)
        add_two_way(read(),read());
    dfs1(1,0),dfs2(1,0);
    for (int i=1;i<=n;i++) if (sum[i]>sum[id]) id=i;
    printf("%d\n",id);
}