【POI2008】【BZOJ1131】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
HINT

Source

树形DP一下
显然把根从一个点的父亲移到这个节点DP值的变化差值是 delta=(nsize[x])size[x]

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 1000010
#define GET (ch>='0'&&ch<='9')
#define LL long long
using namespace std;
int n,top;
int size[MAXN];
LL g[MAXN],f[MAXN];
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
struct edge
{
    int to;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
void insert(int u,int v)    {   e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];  }
void dfs1(int x,int f)
{
    size[x]=1;
    for (edge *i=prev[x];i;i=i->next)
        if (i->to!=f)   dfs1(i->to,x),size[x]+=size[i->to],g[x]+=g[i->to];
    g[x]+=size[x]-1;
}
void dfs2(int x,int F)
{
    f[x]=(F!=0)?(f[F]-size[x]+(n-size[x])):g[x];
    for (edge *i=prev[x];i;i=i->next)   if (i->to!=F)   dfs2(i->to,x);
}
int main()
{
    in(n);int u,v,id;LL maxn=0;
    for (int i=1;i<n;i++)   in(u),in(v),insert(u,v),insert(v,u);
    dfs1(1,0);dfs2(1,0);
    for (int i=1;i<=n;i++)  if (f[i]>maxn)  maxn=f[i],id=i;
    cout<<id<<endl;
}

你可能感兴趣的:(树形DP)