Codeforces600 E. Lomsat gelral（线段树合并）

题意：

给定n个节点的树，每个点有点权，点1为树根
要求计算以每个节点为根的子树中出现次数最多的点权的和（有多个次数最多的则累加，每种点权只加一次）

数据范围：n<=1e5，点权<=n

解法：

可以用权值线段树维护每种点权出现的次数，区间内出现最多的次数，区间内出现最多次数的点权和
每个点的线段树可以由子节点的线段树合并得来，所以可以用线段树合并来做

自底向上建树，记录每个节点对应的线段树根节点，对于第i个点在rt[i]树的区间[1,n]查询出现最多次数的点权和

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e5+5;
vector<int>g[maxm];
int ans[maxm];
int a[maxm];
int n;
//
int lc[maxm*40],rc[maxm*40],ma[maxm*40],sum[maxm*40],cnt[maxm*40];
int r[maxm],tot;
//
void pushup(int k){
    ma[k]=max(ma[lc[k]],ma[rc[k]]);
    sum[k]=0;
    if(ma[lc[k]]==ma[k])sum[k]+=sum[lc[k]];
    if(ma[rc[k]]==ma[k])sum[k]+=sum[rc[k]];
}
void update(int x,int l,int r,int &k){
    if(!k)k=++tot;
    if(l==r){
        cnt[k]=1,ma[k]=1,sum[k]=x;return ;
    }
    int mid=(l+r)/2;
    if(x<=mid)update(x,l,mid,lc[k]);
    else update(x,mid+1,r,rc[k]);
    pushup(k);
}
int merged(int x,int y,int l,int r){
    if(!x||!y)return x+y;
    int k=++tot;//每个节点只需要查询一次,也可以不新建节点,但是这样需要将查询放到dfs里面而不是main里面
    if(l==r){//叶子节点需要合并cnt
        cnt[k]=cnt[x]+cnt[y];
        ma[k]=cnt[k];
        sum[k]=sum[x];
        return k;
    }
    int mid=(l+r)/2;
    lc[k]=merged(lc[x],lc[y],l,mid);
    rc[k]=merged(rc[x],rc[y],mid+1,r);
    pushup(k);
    return k;
}
void dfs(int x,int fa){
    update(a[x],1,n,r[x]);
    for(int v:g[x]){
        if(v==fa)continue;
        dfs(v,x);
        r[x]=merged(r[x],r[v],1,n);
    }
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<n;i++){
        int a,b;cin>>a>>b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    dfs(1,1);
    for(int i=1;i<=n;i++){
        ans[i]=sum[r[i]];//区间[1,n]内的答案就是sum[root],不用写ask()
        cout<<ans[i]<<' ';
    }
    return 0;
}

---