CodeForces - 600E（树上启发式合并）

CF600E
题解：维护两个map，见代码

#include<bits/stdc++.h>
using namespace std; 
typedef long long LL;
const LL maxn=1e5+10;
int n;
LL c[maxn];
map<int,int> cnt[maxn];//cnt[x][i]=ko表示x节点子树颜色为i出现ko次 
map<int,LL> sum[maxn];//sum[x][i]=ko表示x节点子树颜色出现次数为i次的累加权值为ko 
vector<int> G[maxn];
LL ans[maxn];
void dfs(int u,int fa)
{
	cnt[u][c[u]]=1;
	sum[u][1]=c[u];
	int len=G[u].size();
	for(int i=0;i<len;i++){
		int to=G[u][i];
		if(to==fa)continue;
		dfs(to,u);
		if(cnt[u].size()<cnt[to].size()){
			swap(cnt[u],cnt[to]);
			swap(sum[u],sum[to]);
		}
		for(map<int,int>::iterator j=cnt[to].begin();j!=cnt[to].end();j++){
		    int l=j->first,r=j->second;
		    if(cnt[u].count(l)>0){
		        sum[u][cnt[u][l]]-=l;
		        cnt[u][l]+=r;
		        sum[u][cnt[u][l]]+=l;
			}
			else{
			   cnt[u][l]=cnt[to][l];
			   sum[u][cnt[u][l]]+=l;
			}
		}
	}
	map<int,LL>::iterator it=sum[u].end();
    it--;
    ans[u]=it->second;
}
int main() 
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
    	scanf("%lld",&c[i]);
	}
    int s,e;
	for(int i=1;i<=n-1;i++){
		scanf("%d%d",&s,&e);
		G[s].push_back(e);G[e].push_back(s);
	}
	dfs(1,0);
	for(int i=1;i<n;i++){
		printf("%lld ",ans[i]);
	}
	printf("%lld",ans[n]);
	return 0;
}