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

最新推荐文章于 2022-05-03 23:24:45 发布

SC.ldxcaicai

最新推荐文章于 2022-05-03 23:24:45 发布

阅读量407

点赞数 1

分类专栏： # 线段树合并文章标签：数据结构

本文链接：https://blog.csdn.net/dreaming__ldx/article/details/82155540

版权

线段树合并专栏收录该内容

11 篇文章 0 订阅

订阅专栏

该博客介绍了如何解决洛谷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;
}