题目链接:Dominant Indices
解题思路: DSU on Tree 树上启发式合并
讲解: http://codeforces.com/blog/entry/44351
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<vector>
#include<algorithm>
#include<queue>
#include<map>
using namespace std;
typedef pair<int,int> PII;
const int MAXN = 1e6+5;
vector<int> G[MAXN];
int sz[MAXN],dep[MAXN],ans[MAXN],cnt[MAXN];
priority_queue<PII> q;
void dfs1(int u,int fa){
sz[u]=1;
for(auto v:G[u]){
if(v==fa) continue;
dep[v]=dep[u]+1;
dfs1(v,u);
sz[u]+=sz[v];
}
}
void add(int u,int fa){
cnt[dep[u]]++;
q.push({cnt[dep[u]],-dep[u]});
for(auto v:G[u]){
if(v!=fa) add(v,u);
}
}
void del(int u,int fa){
cnt[dep[u]]--;
for(auto v:G[u]){
if(v!=fa) del(v,u);
}
}
void dfs2(int u,int fa,int keep){
int mx = -1,bigChild = -1;
for(auto v:G[u]){
if(v!=fa&&sz[v]>mx){
mx=sz[v],bigChild=v;
}
}
for(auto v:G[u]){
if(v!=fa&&v!=bigChild)
dfs2(v,u,0);
}
if(bigChild!=-1) dfs2(bigChild,u,1);
for(auto v:G[u]){
if(v!=fa&&v!=bigChild) add(v,u);
}
cnt[dep[u]]++;
q.push({cnt[dep[u]],-dep[u]});
ans[u]=-q.top().second-dep[u];
if(keep==0){
del(u,fa);
while(!q.empty()) q.pop();
}
}
void Init(int n){
for(int i=1;i<=n;i++) G[i].clear();
while(!q.empty()) q.pop();
memset(cnt,0,sizeof(cnt));
}
int main(){
int n,u,v;
while(scanf("%d",&n)!=EOF){
Init(n);
for(int i=1;i<=n-1;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs1(1,-1);
dfs2(1,-1,1);
for(int i=1;i<=n;i++){
printf("%d\n",ans[i]);
}
}
return 0;
}