随笔 - 531 文章 - 0 评论 - 3 阅读 - 10215

P4556 [Vani有约会]雨天的尾巴 /【模板】线段树合并

线段树合并板子

100

101

102

103

104

#include<iostream>
#include<cstring>
using namespace std;
const int N =5e5+10,M =N*2;
 
int nxt[M],hd[N],all=1,go[M],n,m;
 int dep[N],f[N][22],father[N];
  
 void add_(int x,int y){
     go[++all]=y,nxt[all]=hd[x],hd[x]=all;
 }
  
  void init(int x,int fa){
     int i,y;
     dep[x]=dep[fa]+1;
     f[x][0]=fa; father[x]=fa; 
      
     for(i=0;i<=19;i++) f[x][i+1]=f[f[x][i]][i];
      
     for(i=hd[x];i;i=nxt[i]){
         y=go[i]; if(y==fa) continue;
         init(y,x);
     }
 }
 int lca(int x,int y){
     int i;
     if(dep[x]<dep[y]) swap(x,y);
      
     for(i=19;i>=0;i--){
         if(dep[f[x][i]]>=dep[y]) x=f[x][i];
     }
     if(x==y) return x;
      
     for(i=19;i>=0;i--){
         if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; 
     }
     return f[x][0];
 }
  
 int rt[N] ;
 int pool=0;
 int New(){  return ++pool;  }
 int sum[N*20],maxp[N*20],ls[N*20],rs[N*20];
  
 void up(int k){
    if(sum[ls[k]]>=sum[rs[k]]) 
        sum[k]=sum[ls[k]],maxp[k]=maxp[ls[k]];
    else
        sum[k]=sum[rs[k]],maxp[k]=maxp[rs[k]];
}
 int change(int k,int l,int r,int x,int v){
    if(k==0) k=New();
     
    if(l==r){
        sum[k]+=v;
        maxp[k]=x; return k;
    }
    int md=(l+r)>>1;
    if(x<=md) ls[k]=change(ls[k],l,md,x,v); 
    else rs[k]=change(rs[k],md+1,r,x,v) ;
     
    up(k);
    return k;
 }
 int merge(int a,int b,int l,int r){
    if (a==0||b==0)
         return a|b;
     
    if(l==r){
        sum[a]+=sum[b]; maxp[a]=l; return a;
    }
     
    int md=(l+r)>>1; 
    ls[a]=merge(ls[a],ls[b],l,md) ; 
    rs[a]=merge(rs[a],rs[b],md+1,r);
    up(a);
    return a;
 }
 int ans[N] ;
  
 void dfs(int x){
    for(int i=hd[x];i;i=nxt[i]){
        int y =go[i] ; if(y==father[x]) continue ;
        dfs(y);
        rt[x]=merge(rt[x],rt[y],1,1e5) ;
    }
    ans[x] = sum[rt[x]]>0 ?maxp[rt[x]]: 0 ;
 }
 signed main(){
    cin>>n>>m;
    for(int i=1,x,y;i<n;i++) cin>>x>>y,add_(x,y),add_(y,x);
    init(1,0) ;
    for(int i=1;i<=m;i++){
        int x,y,z;
        cin>>x>>y>>z; int t =lca(x,y) ;
        rt[x]=change(rt[x],1,1e5,z,1) ; 
        rt[y]=change(rt[y],1,1e5,z,1) ;
        rt[t]=change(rt[t],1,1e5,z,-1);
        rt[father[t]]=change(rt[father[t]],1,1e5,z,-1);
    }
    dfs(1) ;
     
    for(int i=1;i<=n;++i) cout << ans[i] <<endl;
 } 