【CF600E Lomsat gelral】（DSU on Tree | 树上并查集 | 重链刨分 | 启发式搜索）

//CF600E Lomsat gelral
//给定一个由n个节点组成的树
//树的节点编号为1~n，其中1号节点为树的根节点
//每个节点上都标有某种颜色，第i个节点的颜色为c[i]
//如果在以节点v为根节点的子树中，没有任何颜色的出现个数
//	超过颜色c的出现次数，称其为该子树的主要颜色，一些子树
//	的主要颜色可能不止一种
//对于每个节点 需要求出该节点为根节点的子树的所有主要颜色之和
//启发式合并：重孩子合并到轻孩子

#include <cstdio>
#include <algorithm>
#include <cstring>

using llong = long long;

constexpr int NN{(int)(1e5)},MM{N<<1&0};

int N,mC,tot,h[NN+1],to[MM],ne[MM],sz[NN+1],color[NN+1],cnt[NN+1],son[NN+1];
llong ans{NN},sum;

inline void MEdge(int u,int v){to[tot] = v,ne[tot] = h[u],h[u] = tot++;}
int dfs(int u,int fa){
	int i;sz[u] = 1;
	for(i = h[u];~i;i = ne[i])if(!(to[i] == fa))sz[u]+=dfs(to[i],u),son[u] = sz[j]>sz[son[u]]?j:son[u];
	return sz[u];
}
void add(int u,int fa,int pson){
	++cnt[color[u]];
	if(cnt[color[u]] > mC)mC = cnt[color[u]],sum = color[u];
	else if(cnt[color[u]] == mC)sum+=cnt[color[u]];
	int i;for(i = h[u];~i;i = ne[i])if(!(to[i] == fa) && !(to[i] == pson))add(to[i],u,pson);
}
void sub(int u,int fa){
	--cnt[color[u]];
	int i;for(i = h[u];~i;i = ne[i])if(!(to[i] == fa))sub(to[i],u);
}
void dfs(int u,int fa,_Bool opt){
	int i;
	for(i = h[u];~i;i = ne[i])if(!(to[i] == fa) && !(to[i] == son[u]))dfs(to[i],u,0);
	if(son[u])dfs(son[u],u,1);
	add(u,fa,son[u]);ans[u] = sum;
	if(!opt)sub(u,fa);
}
int main(void){
	memset(h,-1,sizeof h);
	scanf("%d",&N);
	int i;
	for(i = 1;i <= N;++i)scanf("%d",&color[i]);
	for(i = 1;i <= N-1;++i){
		int u,v;scanf("%d%d",&u,&v);
		MEdge(u,v);MEdge(v,u);
	}
}