题目大意
统计每个子树里出现次数最多的颜色编号和。
dsu on tree
裸题吧。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
int h[maxn],go[maxn*2],next[maxn*2],c[maxn],cnt[maxn],size[maxn],mx[maxn];
ll ans[maxn];
bool bz[maxn];
int i,j,k,l,t,n,m,tot,num;
ll now;
void add(int x,int y){
go[++tot]=y;
next[tot]=h[x];
h[x]=tot;
}
void travel(int x,int y){
int t=h[x];
size[x]=1;
while (t){
if (go[t]!=y){
travel(go[t],x);
size[x]+=size[go[t]];
}
t=next[t];
}
mx[x]=0;
t=h[x];
while (t){
if (go[t]!=y&&(!mx[x]||size[go[t]]>size[mx[x]])) mx[x]=go[t];
t=next[t];
}
}
void dg(int x,int y,int f){
cnt[c[x]]+=f;
if (cnt[c[x]]>num) num=cnt[c[x]],now=c[x];
else if (cnt[c[x]]==num) now+=(ll)c[x];
int t=h[x];
while (t){
if (go[t]!=y&&!bz[go[t]]) dg(go[t],x,f);
t=next[t];
}
}
void dfs(int x,int y,bool keep){
int t=h[x];
while (t){
if (go[t]!=y&&go[t]!=mx[x]) dfs(go[t],x,0);
t=next[t];
}
if (mx[x]) dfs(mx[x],x,1),bz[mx[x]]=1;
dg(x,y,1);
ans[x]=now;
if (mx[x]) bz[mx[x]]=0;
if (!keep){
dg(x,y,-1);
now=num=0;
}
}
int main(){
scanf("%d",&n);
fo(i,1,n) scanf("%d",&c[i]);
fo(i,1,n-1){
scanf("%d%d",&j,&k);
add(j,k);add(k,j);
}
travel(1,0);
dfs(1,0,0);
fo(i,1,n) printf("%I64d ",ans[i]);
}