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

Posted 2023-06-22 towboa

 

线段树合并板子

 

#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;