樹上啓發式合併與dsu on tree

題目鏈接：https://codeforc.es/contest/600/problem/E

題目大意：給一顆以1爲根的樹，共n個點。點有顏色，顏色從1-n編號。問每顆子樹中出現次數最多的顏色 的編號之和。

啓發式合併做法：

    #include<bits/stdc++.h>
    #include<ext/pb_ds/assoc_container.hpp>
    #include<ext/pb_ds/hash_policy.hpp>
    #define ll long long
    #define rep(i, a, b) for(int i = (a); i <= (b); i++)
    #define pb push_back
    using namespace std;
    using namespace __gnu_pbds;
    const int N = 1e5+1000;
    vector<int>nxt[N];
    int siz[N],son[N];
    void split(int u,int f) {
        son[u] = 0;
        siz[u] = 1;
        for(auto v:nxt[u]) {
            if(v==f)continue;
            split(v,u);
            siz[u] += siz[v];
            if(siz[son[u]]<siz[v]) son[u] = v;
        }
    }
    //----------
    ll val[N],ans[N],p[N];
    unordered_map<ll,ll>num[N],cnt[N]; 
    void add(int u,int f,int t) {
        int col = val[u];
        cnt[t][num[t][col]] -= col;
        num[t][col]++;
        cnt[t][num[t][col]] += col;
        if(num[t][col]>p[t]) p[t]++;
        for(auto v:nxt[u]) {
            if(v==f) continue;
            add(v,u,t);
        }
    }
    void dfs(int u,int f) {
        for(auto v:nxt[u]) {
            if(v==f) continue;
            dfs(v,u);
        }
        if(son[u]) {                    //繼承重兒子的信息
            swap(num[u],num[son[u]]);
            swap(cnt[u],cnt[son[u]]);
            swap(p[u],p[son[u]]);
        }
        for(auto v:nxt[u]) {            
            if(v==f||v==son[u]) continue;
            add(v,u,u);                //暴力合併輕兒子
        }    
        int col = val[u];                //和並這個點本身
        cnt[u][num[u][col]] -= col;
        num[u][col]++;
        cnt[u][num[u][col]] += col;
        if(num[u][col]>p[u]) p[u]++;
        ans[u] = cnt[u][p[u]];
    }
    int main() {
        //ios::sync_with_stdio(0);
        int n;
        cin>>n;
        rep(i, 1, n) cin>>val[i];
        rep(i, 1, n-1) {
            int u,v;
            cin>>u>>v;
            nxt[u].pb(v);
            nxt[v].pb(u);
        }
        split(1,0);
        dfs(1,0);
        rep(i, 1, n) cout<<ans[i]<<' ';
        return 0;
    }

dsu on tree 做法：

#include <bits/stdc++.h>
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
#define mp make_pair
#define pb push_back
#define ll long long
using namespace std;
const int N = 1e5+1000;
vector<int>nxt[N];
int siz[N],son[N];
void split(int u,int f) {
    son[u] = 0;
    siz[u] = 1;
    for(auto v:nxt[u]) {
        if(v==f)continue;
        split(v,u);
        siz[u] += siz[v];
        if(siz[son[u]]<siz[v]) son[u] = v;
    }
}
//---------------------
ll p,num[N],ans[N],Son,cnt[N],val[N];
void add(int u,int f,int data) {
    cnt[num[val[u]]] -= val[u];
    num[val[u]] += data;
    cnt[num[val[u]]] += val[u];
    if(num[val[u]]>p) p++;
    for(auto v:nxt[u]) {
        if(v==f||v==Son) continue;
        add(v,u,data);
    }
}
void dfs(int u,int f,int opt) {
    for(auto v:nxt[u]) {
        if(v==f||v==son[u]) continue;
        dfs(v,u,0);              //計算輕兒子的答案，不保留貢獻
    }
    if(son[u]) dfs(son[u],u,1),Son = son[u];  //計算重兒子的答案，保留貢獻
    add(u,f,1),Son = 0;          //遍歷所有輕兒子，將貢獻合併
    ans[u] = cnt[p];             //記錄答案
    if(!opt) add(u,f,-1),p = 0;  //如果u點是輕兒子，消除全部影響，清空所有信息
}
int main() {
    //freopen("a.txt","r",stdin);
    ios::sync_with_stdio(0);
    int n;
    cin>>n;
    rep(i, 1, n) cin>>val[i];
    rep(i, 1, n-1) {
        int u,v;
        cin>>u>>v;
        nxt[u].pb(v);
        nxt[v].pb(u);
    }
    split(1,0);
    dfs(1,0,0);
    rep(i, 1, n) cout<<ans[i]<<' ';
    return 0;
}