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

传送门

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

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

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

代码:

#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;
}
上一篇:2021-7-5 LeetCode


下一篇:JS验证姓名、邮箱、电话号码