标签:树形DP
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
分析:先默认一个根节点,计算初始的答案,然后每次O(1)计算出其儿子节点作为根节点的ans值,更新答案
Code
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define mem(x,num) memset(x,num,sizeof x)
#define reg(x) for(int i=head[x];i;i=e[i].next)
#define LL long long
using namespace std;
inline LL read()
{LL f=1,x=0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;
}
const int maxn=1e6+6;
LL f[maxn],n,cnt=0,ans=0,head[maxn],deep[maxn],size[maxn];
struct edge{int to,next;}e[maxn<<1];
void dfs1(int x,int fa)
{size[x]=1;f[x]=deep[x];reg(x)if(e[i].to!=fa){deep[e[i].to]=deep[x]+1;dfs1(e[i].to,x);f[x]+=f[e[i].to];size[x]+=size[e[i].to];}
}void dfs2(int x,int fa)
{reg(x)if(e[i].to!=fa){f[e[i].to]=f[x]+n-2*size[e[i].to];dfs2(e[i].to,x);}
}
int main()
{n=read();rep(i,1,n-1){int u=read(),v=read();e[++cnt]=(edge){v,head[u]};head[u]=cnt;e[++cnt]=(edge){u,head[v]};head[v]=cnt;}dfs1(1,0);dfs2(1,0);rep(i,1,n)if(f[i]>f[ans])ans=i;printf("%lld\n",ans);return 0;
}