大意是每个点有一个颜色,询问以每个点为根的子树中出现次数最多的颜色是谁,如果有多个输出编号和。
这个东西用数据结构是不太好做的,考虑优化暴力。
首先最裸的的暴力就是在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;
}