CF1009F-Dominant Indices

题目链接：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;
}