2018.08.28 洛谷P4556 [Vani有约会]雨天的尾巴(树上差分+线段树合并)

该博客介绍了如何解决洛谷P4556问题,通过使用树上差分和线段树合并的方法来维护每个节点出现次数最多颜色的信息。在处理大数据量的颜色时,进行了离散化操作。提供了相关代码实现。
摘要由CSDN通过智能技术生成

传送门
要求维护每个点上出现次数最多的颜色。

对于每次修改,我们用树上差分的思想,然后线段树合并统计答案就行了。

注意颜色很大需要离散化。

代码:

#include<bits/stdc++.h>
#define N 100005
#define Max 100000
using namespace std;
inline int read(){
    int ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+ch-'0',ch=getchar();
    return ans;
}
int n,m,first[N],ans[N],cnt=0,tot=0,rt[N],son[N*60][2],mx[N*60],top[N],fa[N],dep[N],siz[N],hson[N],mp[N],num[N],sig=0;
struct Q{int x,y,z,w;}q[N];
struct edge{int v,next;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;}
inline void dfs1(int p){
    siz[p]=1,hson[p]=0;
    for(int i=first[p];i;i=e[i].next){
        int v=e[i].v;
        if(v==fa[p])continue;
        fa[v]=p,dep[v]=dep[p]+1,dfs1(v),siz[p]+=siz[v];
        if(siz[v]>siz[hson[p]])hson[p]=v;
    }
}
inline void dfs2(int p,int tp){
    top[p]=tp;
    if(hson[p])dfs2(hson[p],tp);
    for(int i=first[p];i;i=e[i].next){
        int v=e[i].v;
        if(v!=fa[p]&&v!=hson[p])dfs2(v,v);
    }
}
inline int lca(int x,int y){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        x=fa[top[x]];
    }
    return dep[x]<dep[y]?x:y;
}
inline int max(int a,int b){return a>b?a:b;}
inline void modify(int&p,int l,int r,int k,int v){
    if(!p)p=++tot;
    if(l==r){mx[p]+=v;return;}
    int mid=l+r>>1;
    if(k<=mid)modify(son[p][0],l,mid,k,v);
    else modify(son[p][1],mid+1,r,k,v);
    mx[p]=max(mx[son[p][0]],mx[son[p][1]]);
}
inline int merge(int x,int y,int l,int r){
    if(!x||!y)return x+y;
    if(l==r){mx[x]+=mx[y];return x;}
    int mid=l+r>>1;
    son[x][0]=merge(son[x][0],son[y][0],l,mid);
    son[x][1]=merge(son[x][1],son[y][1],mid+1,r);
    mx[x]=max(mx[son[x][0]],mx[son[x][1]]);
    return x;
}
inline int query(int p,int l,int r){
    if(l==r){
        if(!mx[p])return 0;
        return mp[l];
    }
    int mid=l+r>>1;
    if(mx[son[p][0]]>=mx[son[p][1]])return query(son[p][0],l,mid);
    return query(son[p][1],mid+1,r);
}
inline void dfs(int p){
    for(int i=first[p];i;i=e[i].next){
        int v=e[i].v;
        if(v==fa[p])continue;
        dfs(v),rt[p]=merge(rt[p],rt[v],1,sig);
    }
    ans[p]=query(rt[p],1,sig);
}
int main(){
    n=read(),m=read();
    for(int i=1;i<n;++i){
        int u=read(),v=read();
        add(u,v),add(v,u); 
    }
    dfs1(1),dfs2(1,1);
    for(int i=1;i<=m;++i)q[i].x=read(),q[i].y=read(),q[i].z=lca(q[i].x,q[i].y),num[++sig]=q[i].w=read();
    sort(num+1,num+sig+1);
    sig=unique(num+1,num+sig+1)-num-1;
    for(int i=1;i<=m;++i){
        int tmp=lower_bound(num+1,num+sig+1,q[i].w)-num;
        mp[tmp]=q[i].w;
        modify(rt[q[i].x],1,sig,tmp,1),modify(rt[q[i].y],1,sig,tmp,1);
        modify(rt[q[i].z],1,sig,tmp,-1),modify(rt[fa[q[i].z]],1,sig,tmp,-1);
    }
    dfs(1);
    for(int i=1;i<=n;++i)cout<<ans[i]<<'\n';
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值