[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<<][],v1[M<<];
 long long v2[M<<],ans[M];
 struct point{int to,next;}e[M<<];
 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]=,v2[node]=x;return;
     }int mid=(l+r)/;
     if(x<=mid) insert(ls,l,mid,x);
     else insert(rs,mid+,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)/;
     ch[node][]=merge(ch[x][],ch[y][],l,mid);
     ch[node][]=merge(ch[x][],ch[y][],mid+,r);
     update(node);return node;
 }
 void dfs(int x,int fa) {
     insert(rt[x],,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],,n);
     }
     ans[x]=v2[rt[x]];
 }
 int main() {
     scanf("%d",&n);
     for(int i=;i<=n;i++) scanf("%d",&co[i]);
     for(int i=;i<n;i++) {
         int x,y;scanf("%d%d",&x,&y);
         add(x,y),add(y,x);
     }
     dfs(,);
     for(int i=;i<=n;i++) printf("%lld ",ans[i]);
     return ;
 }