CF600E Lomsat gelral——线段树合并

题目描述

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

　　这个题意是真的窒息。。。具体意思是说，每个节点有一个颜色，你要找的是每个子树中颜色的众数（可能有多个），比如子树中有$3个2,3个1，3个5，那么2,1,5都是众数，答案为2+1+5=8$。

思路

　　线段树合并。权值线段树覆盖颜色$1->100000，用sum$表示颜色最多出现的次数，$ans$表示答案。分$3种情况pushup$即可。

1. 左右子树$sum$相等
2. 左边$>$右边
3. 左边$<$右边

　　$dfs的时候merge$一下即可。

code

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector> 
#define smid (l+r>>1)
#define I inline 
using namespace std;
typedef long long LL;
const int N=100010;
LL col[N];
LL maxcol;
int n;
int ls[N*40],rs[N*40],cnt,rt[N];
LL sum[N*40],ans[N*40];
vector<int>g[N];
LL out[N];

I void pushup(int now)
{
    if(sum[ls[now]]==sum[rs[now]])
    {
        sum[now]=sum[ls[now]];
        ans[now]=ans[ls[now]]+ans[rs[now]];
    }
    else if(sum[ls[now]]<sum[rs[now]])
    {
        sum[now]=sum[rs[now]];
        ans[now]=ans[rs[now]];
    }
    else
    {
        sum[now]=sum[ls[now]];
        ans[now]=ans[ls[now]];
    }
}

I void modify(int &now,int l,int r,int pos)
{
    if(!now)now=++cnt;
    if(l==r)
    {
        sum[now]++;ans[now]=l;
        return;
    }
    if(pos<=smid)modify(ls[now],l,smid,pos);
    else modify(rs[now],smid+1,r,pos);
    pushup(now);
}

I int merge(int x,int y,int l,int r)
{
    if(!x||!y)return x+y;
    if(l==r)
    {
        sum[x]+=sum[y];ans[x]=l;
        return x;
    }
    ls[x]=merge(ls[x],ls[y],l,smid);
    rs[x]=merge(rs[x],rs[y],smid+1,r);
    pushup(x);
    return x;
}

I void dfs(int u,int fa)
{
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v==fa)continue;
        dfs(v,u);
        merge(rt[u],rt[v],1,100000);
    }
    modify(rt[u],1,100000,col[u]);
    out[u]=ans[rt[u]];
}

int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>col[i];
        rt[i]=i;cnt++;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;cin>>x>>y;
        g[x].push_back(y);g[y].push_back(x);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++)
    {
        cout<<out[i]<<" ";
    }
}

洛谷上交不了，必须到$CF$上交，但是$CF$上不给用scanf("%lld")，就加了$cin$加速。