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=(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 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;
}