bzoj1131: [POI2008]Sta

时间:2021-09-21 21:46:11

思路:首先先求出以1为根的答案,然后考虑由i转移到i的儿子的答案的变化,显然以son[i]为根的子树的所有结点的深度都会减一,其余的点的深度都会加一,然后就可以直接O(n)求出所有结点的答案,然后取max更新答案即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define maxn 1000005 int n,tot;
int now[maxn],pre[*maxn],son[*maxn],dep[maxn],size[maxn];
long long dp[maxn]; int read(){
int x=;char ch=getchar();
for (;ch<''||ch>'';ch=getchar());
for (;ch>=''&&ch<='';ch=getchar()) x=x*+ch-'';
return x;
} void add(int a,int b){
son[++tot]=b;
pre[tot]=now[a];
now[a]=tot;
} void link(int a,int b){
add(a,b),add(b,a);
} void dfs(int x,int fa){
dep[x]=dep[fa]+,dp[x]=dep[x],size[x]=;
for (int p=now[x];p;p=pre[p])
if (son[p]!=fa){
dfs(son[p],x);
dp[x]+=dp[son[p]];
size[x]+=size[son[p]];
}
} void tree_dp(int x,int fa){
for (int p=now[x];p;p=pre[p])
if (son[p]!=fa){
dp[son[p]]=dp[x]+n-*size[son[p]];
tree_dp(son[p],x);
}
} int main(){
n=read();
for (int i=,a,b;i<n;i++) a=read(),b=read(),link(a,b);
dfs(,),tree_dp(,);int ans=;
for (int i=;i<=n;i++) if (dp[ans]<dp[i]) ans=i;
printf("%d\n",ans);
return ;
}