[CF600E]Lomsat gelral

题目大意

统计每个子树里出现次数最多的颜色编号和。

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]);}