【POI2008】【BZOJ1131】Sta

来源：互联网发布：网络写手怎么赚钱编辑：程序博客网时间：2024/06/05 09:50

Description

给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大

Input

给出一个数字N,代表有N个点.N<=1000000 下面N-1条边.

Output

输出你所找到的点,如果具有多个解,请输出编号最小的那个.

Sample Input

1 4

5 6

4 5

6 7

6 8

2 4

3 4
Sample Output

7
HINT

Source

树形DP一下
显然把根从一个点的父亲移到这个节点DP值的变化差值是delta=(n−size[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 longusing 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;}

1 0