题目本身并不难,POI的题真是又经典又好。
本题卡vector 不开快读的vector存图很容易被卡 本人就被卡了人傻自带大常数
题目本身要求在无根树中选一个节点作根,使形成的有根树所有结点深度和最大。
考虑到 对于相邻的两个结点 转移是O(1)的,假设结点u的答案是H[u],其子节点v从H[u]到H[v]只需 考虑转移后对v的子树 和剩余结点即可。
H[v]=H[u]+n-2*sz[v];
补上完整代码
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
const int maxn = 2e6+500;
int x,y,n,cnt;
long long H[maxn];
int deep[maxn],sz[maxn],head[maxn];
struct edge{
int to,next;
} e[maxn];
void add(int u,int v)
{
e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
e[++cnt].to=u;e[cnt].next=head[v];head[v]=cnt;
}
void dfs1(int u,int fa,int dep)
{
deep[u]=dep;sz[u]=1;
H[u]=deep[u];
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(v==fa) continue;
dfs1(v,u,dep+1);
H[u]+=H[v];
sz[u]+=sz[v];
}
}
void dp(int u,int fa)
{
for(int i=head[u];i!=-1;i=e[i].next)
{
int v=e[i].to;
if(v==fa) continue;
H[v]=H[u]+n-2*sz[v];
dp(v,u);
}
}
int main()
{
scanf("%d",&n);
memset(head,-1,sizeof(head));
for(int i=1;i<n;i++)
{
scanf("%d%d",&x,&y);
add(x,y);
}
dfs1(1,-1,0);
dp(1,-1);
int mx=0;
for(int i=1;i<=n;i++)
{
if(H[i]>H[mx]) mx=i;
}
cout<<mx<<endl;
return 0;
}