dsu on tree 启发式树上合并

解决树上关于子树信息的静态问题，如某节点子树中符合某条件的个数。
cf-600E
给定一棵树，每个节点有一个颜色，求节点x的子树中出现次数最多的颜色，如果有相同次数的，颜色id相加。
考虑暴力，遍历x的每个节点，统计，若只用一个数组，那么两个子节点之间必须要清空数组。
优化，对于清空操作，选择x的轻儿子进行清空，重儿子最后遍历，不需要清空，然后将信息合并给x。

#include<bits/stdc++.h>
using namespace std;
char buf[1<<20],*P1=buf,*P2=buf;
#define gc() (P1==P2&&(P2=(P1=buf)+fread(buf,1,1<<20,stdin),P1==P2)?EOF:*P1++)
#define TT template<class T>inline
TT bool read(T &x){
    x=0;char c=gc();bool f=0;
    while(c<48||c>57){if(c==EOF)return 0;f^=(c=='-'),c=gc();}
    while(47<c&&c<58)x=(x<<3)+(x<<1)+(c^48),c=gc();
    if(f)x=-x;return 1;
}
TT bool read(T&a,T&b){return read(a)&&read(b);}
TT bool read(T&a,T&b,T&c){return read(a)&&read(b)&&read(c);}
typedef long long ll;
const ll MAXN=1e5+8,mod=1e9+8,inf=1ll<<50;
#define lowbit(x) (x&(-x))
#define Max(a,b) if(b>a)a=b
#define Min(a,b) if(b<a)a=b
struct E{int y,nt;}e[MAXN<<1];
int head[MAXN],cnt;
inline void add(int x,int y){//x->y
    e[++cnt].y=y;
    e[cnt].nt=head[x];
    head[x]=cnt;
}
int n,c[MAXN];
int tot[MAXN],son[MAXN],fa[MAXN];
int dfs_id[MAXN],id_cnt,dfs_val[MAXN];
//某个节点的dfs序，dfs序代表的颜色。
void dfs1(int x){
    tot[x]=1;
    dfs_id[x]=++id_cnt;
    dfs_val[id_cnt]=c[x];
    for(int i=head[x];i;i=e[i].nt){
        int y=e[i].y;
        if(tot[y])continue;
        fa[y]=x;
        dfs1(y);
        tot[x]+=tot[y];
        if(!son[x]||tot[y]>tot[son[x]])son[x]=y;
    }
}
ll col_num[MAXN],max_num,num_col[MAXN];
//开ll，爆int
void delans(int x){//删除x节点的子树。
    for(int i=dfs_id[x];i<dfs_id[x]+tot[x];++i){
        int now_col=dfs_val[i];
        num_col[col_num[now_col]--]-=now_col;
        while(max_num>0&&num_col[max_num]==0)max_num--;
    }
}
void addans(int x){//加上x节点的子树
    for(int i=dfs_id[x];i<dfs_id[x]+tot[x];++i){
        int now_col=dfs_val[i];
        num_col[++col_num[now_col]]+=now_col;
        Max(max_num,col_num[now_col]);
    }
}
ll ans[MAXN];
void dfs2(int x){
    for(int i=head[x];i;i=e[i].nt){
        int y=e[i].y;
        if(y==son[x]||y==fa[x])continue;
        dfs2(y);//y是轻儿子，遍历后删除
        delans(y);
    }
    if(son[x])dfs2(son[x]);
    for(int i=head[x];i;i=e[i].nt){
        int y=e[i].y;
        if(y==son[x]||y==fa[x])continue;
        addans(y);//把轻儿子加上，
    }
    num_col[++col_num[c[x]]]+=c[x];
    Max(max_num,col_num[c[x]]);
    ans[x]=num_col[max_num];//更新答案。
}
int main() {
    read(n);
    for(int i=1;i<=n;++i)read(c[i]);
    for(int i=1,x,y;i<n;++i){
        read(x,y);
        add(x,y);
        add(y,x);
    }
    dfs1(1);
    dfs2(1);
    for(int i=1;i<=n;++i){
        printf("%I64d ",ans[i]);
    }
    return 0;
}