bzoj 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

转移很明显

但是想了会加加减减。。

所以 大概就是有当前节点转到儿子节点

所以分为 儿子节点的上面和 下面 然后没 了╮(╯_╰)╭

总结：当状态或者转移明显时 考虑动态规划 当然书上的就树形喽

记得开long long

#include<cstdio>
#include<algorithm>
#include<cstring>
#define int long long
//by mars_ch
using namespace std;
struct data
{
    int f,t,nxt;
}e[1000005*2];
int first[1000005],tot;
void add(int a,int b)
{
    e[tot].f=a,e[tot].t=b,e[tot].nxt=first[a];
    first[a]=tot++;
 } 
int n,sz[1000005],dep[1000005],tod[1000005];
void dfs1(int x,int fa)
{
    sz[x]=1;
    for(int i=first[x];i!=-1;i=e[i].nxt)
    {
        int t=e[i].t;
        if(t == fa) continue;
        dep[t]=dep[x]+1;
        dfs1(t,x);
        sz[x]+=sz[t];
        tod[x]=tod[t]+sz[x];
    }
 } 
 void dfs2(int x,int fa)
 {
    if(x!=1)tod[x]=tod[fa]+n-sz[x]-sz[x];
    for(int i=first[x];i!=-1;i=e[i].nxt)
    {
        int t=e[i].t;
        if(t == fa) continue;
        dfs2(t,x);
     }
 }
signed main()
{
    scanf("%lld",&n);
    memset(first,-1,sizeof(first));
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        scanf("%lld%lld",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs1(1,0);
    dfs2(1,0);
    int ans=0,maxx=0;
    for(int i=1;i<=n;i++)
    {
        if(maxx<tod[i])
        {
            maxx=tod[i];
            ans=i;
        }
    }
    printf("%lld\n",ans);
    return 0;
}