题目大意:给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大
题解:比较套路的树形dp,自顶向下更新答案
首先dfs得到以1为根的答案
考虑将选择的点由1移动到1的儿子v
有size[v]个点dep-1,(n-size[v])个点dep+1
即f[v]=f[x]+n−2∗size[v] 即 f [ v ] = f [ x ] + n − 2 ∗ s i z e [ v ]
我的收获:树形dp的顺序
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <ctime>
#include <cmath>
using namespace std;
const int M=1000005;
#define INF 0x7fffffff
int n;
int t,head[M];
int dep[M],size[M];
long long f[M];
struct edge{int to,nex;}e[M<<1];
void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}
int read()
{
int x=0,f=1;char c=getchar();
while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
while(c<='9'&&c>='0') x=x*10+c-48,c=getchar();
return x*f;
}
void dfs(int x,int fa)
{
size[x]=1;f[x]=dep[x];
for(int i=head[x];i!=-1;i=e[i].nex){
int v=e[i].to;
if(v==fa) continue;
dep[v]=dep[x]+1,dfs(v,x);
size[x]+=size[v],f[x]+=f[v];
}
}
void dfsans(int x,int fa)
{
for(int i=head[x];i!=-1;i=e[i].nex){
int v=e[i].to;
if(v==fa) continue;
f[v]=f[x]+n-2*size[v];
dfsans(v,x);
}
}
void work()
{
dfs(1,0);dfsans(1,0);
int root=0;
for(int i=1;i<=n;i++) if(f[i]>f[root]) root=i;
cout<<root<<endl;
}
void init()
{
n=read();int x,y;
memset(head,-1,sizeof(head));
for(int i=1;i<n;i++) x=read(),y=read(),add(x,y),add(y,x);
}
int main()
{
init();
work();
return 0;
}