P4556 [Vani有约会]雨天的尾巴（线段树合并+lca）

P4556 [Vani有约会]雨天的尾巴

每个操作拆成4个进行树上差分，动态开点线段树维护每个点的操作。

离线处理完向上合并就好了

luogu倍增lca被卡了5分.....于是用rmq维护....

常数很大，被bzoj卡了（但是我不想改了）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#define ri register int
using namespace std;
int read(){
    char c=getchar(); int x=;
    while(c<''||c>'') c=getchar();
    while(''<=c&&c<='') x=x*+c-,c=getchar();
    return x;
}
#define N 100001
#define W 5000005
int n,m,fa[N],fir[N],Log[N*],f[][N*],tp,dfn[N],cc,ans[N];//一定记住st表的大小是n*2
int u,rt[N],lc[W],rc[W],mx[W],id[W];
vector <int> g[N];
#define mid (l+r)/2
void up(int o){
    mx[o]=id[o]=;
    if(mx[lc[o]]>mx[o]) mx[o]=mx[lc[o]],id[o]=id[lc[o]];
    if(mx[rc[o]]>mx[o]) mx[o]=mx[rc[o]],id[o]=id[rc[o]];
}
void merge(int &o,int p,int l,int r){//线段树合并
    if(!o||!p){o=o+p; return;}
    if(l==r){mx[o]+=mx[p]; return;}
    merge(lc[o],lc[p],l,mid);
    merge(rc[o],rc[p],mid+,r); up(o);
}
void ins(int &o,int l,int r,int k,int v){
    if(!o)o=++u;
    if(l==r){mx[o]+=v,id[o]=l; return;}
    if(k<=mid) ins(lc[o],l,mid,k,v);
    else ins(rc[o],mid+,r,k,v);
    up(o);
}
void dfs(int x,int Fa){
    fa[x]=Fa; f[][++tp]=dfn[++cc]=x; fir[x]=tp;
    for(int i=;i<g[x].size();++i) if(g[x][i]!=Fa) dfs(g[x][i],x),f[][++tp]=x;
}
inline int Min(int x,int y){return fir[x]<fir[y]?x:y;}
int lca(int x,int y){
    ri l=fir[x],r=fir[y]; if(l>r) swap(l,r);
    ri k=Log[r-l+];
    return Min(f[k][l],f[k][r-(<<k)+]);
}
int main(){
    n=read(); m=read(); int u,v,w,p;
    for(ri i=;i<n;++i){
        u=read(),v=read();
        g[u].push_back(v);
        g[v].push_back(u);
    }dfs(,); Log[]=-;
    for(ri i=;i<=tp;++i) Log[i]=Log[i>>]+;
    for(ri i=;i<=Log[tp];++i)
        for(ri j=;j+(<<i)-<=tp;++j)
            f[i][j]=Min(f[i-][j],f[i-][j+(<<(i-))]);
    while(m--){
        u=read(),v=read(),w=read(); p=lca(u,v);
        ins(rt[u],,N-,w,);
        ins(rt[v],,N-,w,);
        ins(rt[p],,N-,w,-);
        if(fa[p]) ins(rt[fa[p]],,N-,w,-);//拆成4个操作
    }
    for(ri i=n;i;--i)
        p=dfn[i],ans[p]=id[rt[p]],merge(rt[fa[p]],rt[p],,N-);
    for(ri i=;i<=n;++i) printf("%d\n",ans[i]);
    return ;
}