[CF600E]Lomsat gelral

题意翻译

一棵树有n个结点，每个结点都是一种颜色，每个颜色有一个编号，求树中每个子树的最多的颜色编号的和。

线段树合并板子题，没啥难度，注意开long long

不过这题$dsu$ $on$ $tree$确实更快

代码：

#include<iostream>
#include<cstdio>
#define ls ch[node][0]
#define rs ch[node][1]
#define M 100010
using namespace std;
int n,m,num,cnt;
int head[M],rt[M],co[M];
int ch[M<<7][2],v1[M<<7];
long long v2[M<<7],ans[M];
struct point{int to,next;}e[M<<1];
void add(int from,int to) {
    e[++num].next=head[from];
    e[num].to=to;
    head[from]=num;
}
void update(int node) {
    if(v1[ls]==v1[rs]) {
        v1[node]=v1[ls];v2[node]=v2[ls]+v2[rs];
    }
    else{
        if(v1[ls]>v1[rs]) v1[node]=v1[ls],v2[node]=v2[ls];
        else v1[node]=v1[rs],v2[node]=v2[rs];
    }
}
void insert(int &node,int l,int r,int x) {
    if(!node) node=++cnt;
    if(l==r) {
        v1[node]=1,v2[node]=x;return;
    }int mid=(l+r)/2;
    if(x<=mid) insert(ls,l,mid,x);
    else insert(rs,mid+1,r,x);
    update(node);
}
int merge(int x,int y,int l,int r) {
    if(!x||!y) return x+y;
    int node=++cnt;
    if(l==r) {
        v1[node]=v1[x]+v1[y];v2[node]=l;return node;
    }
    int mid=(l+r)/2;
    ch[node][0]=merge(ch[x][0],ch[y][0],l,mid);
    ch[node][1]=merge(ch[x][1],ch[y][1],mid+1,r);
    update(node);return node;
}
void dfs(int x,int fa) {
    insert(rt[x],1,n,co[x]);
    for(int i=head[x];i;i=e[i].next) {
        int to=e[i].to;
        if(to==fa) continue;
        dfs(to,x);
        rt[x]=merge(rt[x],rt[to],1,n);
    }
    ans[x]=v2[rt[x]];
}
int main() {
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&co[i]);
    for(int i=1;i<n;i++) {
        int x,y;scanf("%d%d",&x,&y);
        add(x,y),add(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++) printf("%lld ",ans[i]);
    return 0;
}