[Coderforces600E] Lomsat gelral

大意是每个点有一个颜色，询问以每个点为根的子树中出现次数最多的颜色是谁，如果有多个输出编号和。

这个东西用数据结构是不太好做的，考虑优化暴力。

首先最裸的的暴力就是在dfs序上跑莫队，但因为是树的结构所以有更优美的暴力。

参考的博客：http://www.cnblogs.com/candy99/p/dsuontree.html

类似树上启发式合并，叫做dsu on tree。

0.首先轻重链剖分。

1.然后先处理轻儿子，处理完把贡献删除。

2.处理重儿子，保留贡献。

3.把轻儿子的贡献再加上。

4.算这个点的答案。

5.如果这个点是轻儿子就把数组里的东西清空。

因为每个点到根只有$log$条轻链，所以在执行3和5时每个点最多加入删除$log$次。

复杂度$nlogn$

#include<bits/stdc++.h>
#define ll long long
#define N 100005
using namespace std;
int v[N],head[N],ver[N*2],nxt[N*2],tot;
void add(int a,int b)
{
	tot++;nxt[tot]=head[a];head[a]=tot;ver[tot]=b;return;
}
int n,size[N],son[N];
void dfs(int x,int f)
{
	size[x]=1;
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f)continue;
		dfs(ver[i],x);
		size[x]+=size[ver[i]];
		if(size[ver[i]]>size[son[x]])son[x]=ver[i];
	}
}
int cnt[N];
ll sum[N],ans[N];
int mx;
void ds(int x,int f,int z)
{
	sum[cnt[v[x]]]-=v[x];
	cnt[v[x]]+=z;
	sum[cnt[v[x]]]+=v[x];
	if(cnt[v[x]]>mx)mx=cnt[v[x]];
	else if(sum[mx]==0)mx--;
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f)continue;
		ds(ver[i],x,z);
	}
}
void dffs(int x,int f,int op)
{
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f||ver[i]==son[x])continue;
		dffs(ver[i],x,0);
	}
	if(son[x])dffs(son[x],x,1);
	sum[cnt[v[x]]]-=v[x];
	cnt[v[x]]++;
	sum[cnt[v[x]]]+=v[x];
	if(cnt[v[x]]>mx)mx=cnt[v[x]];
	for(int i=head[x];i;i=nxt[i])
	{
		if(ver[i]==f||ver[i]==son[x])continue;
		ds(ver[i],x,1);
	}
	ans[x]=sum[mx];
	if(!op)ds(x,f,-1);
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&v[i]);
	int t1,t2;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&t1,&t2);
		add(t1,t2);add(t2,t1);
	}
	dfs(1,-1);
	dffs(1,-1,0);
	for(int i=1;i<=n;i++)
	{
		printf("%I64d ",ans[i]);
	}
	return 0;
}