【Codeforces】600E Lomsat gelral (dsu on tree)

题目传送门

优美的暴力什么的太暴力了吧……

首先我们证明一下dsu on tree的复杂度，我们挑出一个节点的重儿子，就是树剖里的重儿子，把所有的轻儿子的信息加入到重儿子里，每次的树的大小至少翻倍，所以时间复杂度为$O(n\log_2n)$。

dsu on tree也有一定的局限性：1.只能支持子树查询；2.不支持修改修改。

个人感觉dsu on tree就是处理树上的轻重儿子关系吧，就是在轻儿子上多做修改，时间复杂度肯定比在重儿子上做修改要小。

回到题目上，题目大意：给定一棵树，每个节点有颜色c[i]，定义子树u中出现次数最多的颜色为u的主要颜色(可以有多个)，求每个节点的主要颜色数值和。

发现这题就是一道裸题，直接上dsu on tree就行了。

附上AC代码：

#include <cstdio>
#include <cctype>
using namespace std;

typedef long long ll;
const int N=1e5+10;
struct side{
    int to,nt;
}s[N<<1];
int n,c[N],h[N],num,x,y,f[N],sz[N],hs[N],a[N],mx;
bool b[N];
ll ans[N],sum;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline void add(int x,int y){
    s[++num]=(side){y,h[x]},h[x]=num;
    s[++num]=(side){x,h[y]},h[y]=num;
}

inline void so1(int x,int fa){
    f[x]=fa,sz[x]=1;
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=fa){
            so1(s[i].to,x),sz[x]+=sz[s[i].to];
            if (sz[s[i].to]>sz[hs[x]]) hs[x]=s[i].to;
        }
    return;
}

inline void change(int x,int w){
    a[c[x]]+=w;
    if (w>0&&a[c[x]]>=mx){
        if (a[c[x]]>mx) sum=0,mx=a[c[x]];
        sum+=c[x];
    }
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]&&!b[s[i].to]) change(s[i].to,w);
    return;
}

inline void so2(int x,int mk){
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=f[x]&&s[i].to!=hs[x]) so2(s[i].to,0);
    if (hs[x]) so2(hs[x],1),b[hs[x]]=1;
    change(x,1),ans[x]=sum;
    if (hs[x]) b[hs[x]]=0;
    if (!mk) change(x,-1),mx=sum=0;
    return;
}

int main(void){
    read(n);
    for (int i=1; i<=n; ++i) read(c[i]);
    for (int i=1; i<n; ++i) read(x),read(y),add(x,y);
    so1(1,0),so2(1,0);
    for (int i=1; i<=n; ++i) printf("%lld ",ans[i]);
    return 0;
}