【Codeforces】600E Lomsat gelral【DSU on tree】

【题目描述】
传送门

【题解】

DSU板子，直接看我博客

代码如下

#include<cstdio>
#include<cctype>
#define MAXN 100005
#define LL __int64
using namespace std;
int n,AnsMax,a[MAXN];
LL c,Ans[MAXN];
bool vis[MAXN];
int hsh[MAXN],Siz[MAXN],Son[MAXN],Fa[MAXN];
struct Edge{
    int tot,lnk[MAXN],son[MAXN<<1],nxt[MAXN<<1];
    void Add(int x,int y){son[++tot]=y;nxt[tot]=lnk[x];lnk[x]=tot;}
}E;
int read(){
    int ret=0;char ch=getchar();bool f=1;
    for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
    for(; isdigit(ch);ch=getchar()) ret=(ret<<3)+(ret<<1)+ch-48;
    return f?ret:-ret;
}
void First(int x,int f){
    Fa[x]=f;Siz[x]=1; 
    for(int j=E.lnk[x],SizeMax=0;j;j=E.nxt[j])
    if(E.son[j]!=f){
        First(E.son[j],x);Siz[x]+=Siz[E.son[j]];
        if(Siz[E.son[j]]>SizeMax) SizeMax=Siz[E.son[j]],Son[x]=E.son[j];
    }
}
void Cal(int x,int p){
    hsh[a[x]]+=p;
    if(p>0&&hsh[a[x]]==AnsMax) c+=a[x];
    if(p>0&&hsh[a[x]]>AnsMax) AnsMax=hsh[a[x]],c=a[x];
    for(int j=E.lnk[x];j;j=E.nxt[j])
    if(E.son[j]!=Fa[x]&&!vis[E.son[j]]) Cal(E.son[j],p);
}
void Second(int x,int T){
    for(int j=E.lnk[x];j;j=E.nxt[j])
    if(E.son[j]!=Fa[x]&&E.son[j]!=Son[x]) Second(E.son[j],0);
    if(Son[x]) Second(Son[x],1),vis[Son[x]]=1;
    Cal(x,1);Ans[x]=c;
    if(Son[x]) vis[Son[x]]=0;
    if(!T) Cal(x,-1),c=AnsMax=0;
}
int main(){
    #ifndef ONLINE_JUDGE
    freopen("prob.in","r",stdin);
    freopen("prob.out","w",stdout);
    #endif
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<n;i++){int x=read(),y=read();E.Add(x,y);E.Add(y,x);}
    First(1,0);Second(1,0);
    for(int i=1;i<=n;i++) printf(i==n?"%I64d\n":"%I64d ",Ans[i]);
    return 0;
}