CF600E:Lomsat gelral(线段树合并)

Description

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

Input

第一行一个$n$。第二行$n$个数字是$c[i]$。后面$n-1$行给出树边。

Output

一行答案。

Sample Input1

4
1 2 3 4
1 2
2 3
2 4

Sample Output1

10 9 3 4

Sample Input2

15
1 2 3 1 2 3 3 1 1 3 2 2 1 2 3
1 2
1 3
1 4
1 14
1 15
2 5
2 6
2 7
3 8
3 9
3 10
4 11
4 12
4 13

Sample Output2

6 5 4 3 2 3 3 1 1 3 2 2 1 2 3

Solution

线段树合并模板题，但我竟然读错题了QAQ……

情况特殊所以$Merge$的时候要加上$l$，$r$。

没啥好说的看代码就能懂……

Code

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #define N (100009)
 #define LL long long
 using namespace std;

 struct Sgt{int ls,rs,max; LL val;}Segt[N*];
 struct Edge{int to,next;}edge[N<<];
 int n,x,u,v,sgt_num,c[N],Root[N];
 int head[N],num_edge;

 void add(int u,int v)
 {
     edge[++num_edge].to=v;
     edge[num_edge].next=head[u];
     head[u]=num_edge;
 }

 void Pushup(int now)
 {
     int ls=Segt[now].ls, rs=Segt[now].rs;
     if (Segt[ls].max==Segt[rs].max)
     {
         Segt[now].max=Segt[ls].max;
         Segt[now].val=Segt[ls].val+Segt[rs].val;
     }
     else if (Segt[ls].max>Segt[rs].max)
     {
         Segt[now].max=Segt[ls].max;
         Segt[now].val=Segt[ls].val;
     }
     else
     {
         Segt[now].max=Segt[rs].max;
         Segt[now].val=Segt[rs].val;
     }
 }

 void Update(int &now,int l,int r,int v)
 {
     if (!now) now=++sgt_num;
     if (l==r)
     {
         Segt[now].max=;
         Segt[now].val=v;
         return;
     }
     int mid=(l+r)>>;
     if (v<=mid) Update(Segt[now].ls,l,mid,v);
     else Update(Segt[now].rs,mid+,r,v);
     Pushup(now);
 }

 int Merge(int x,int y,int l,int r)
 {
     if (!x || !y) {return x|y;}
     int tmp=++sgt_num;
     if (l==r)
     {
         Segt[tmp].max=Segt[x].max+Segt[y].max;
         Segt[tmp].val=l; return tmp;
     }
     int mid=(l+r)>>;
     Segt[tmp].ls=Merge(Segt[x].ls,Segt[y].ls,l,mid);
     Segt[tmp].rs=Merge(Segt[x].rs,Segt[y].rs,mid+,r);
     Pushup(tmp); return tmp;
 }

 void DFS(int x,int fa)
 {
     for (int i=head[x]; i; i=edge[i].next)
         if (edge[i].to!=fa)
         {
             DFS(edge[i].to,x);
             Root[x]=Merge(Root[x],Root[edge[i].to],,n);
         }
 }

 int main()
 {
     scanf("%d",&n);
     for (int i=; i<=n; ++i)
     {
         scanf("%d",&c[i]);
         Update(Root[i],,n,c[i]);
     }
     for (int i=; i<=n-; ++i)
     {
         scanf("%d%d",&u,&v);
         add(u,v); add(v,u);
     }
     DFS(,);
     for (int i=; i<=n; ++i)
         printf("%lld ",Segt[Root[i]].val);
 }