bzoj4538 [Hnoi2016]网络（树链剖分，线段树套堆）

给定一棵树，三种操作：
1、加入一条路径
2、删除一条已加入的路径
3、询问不过一个点x的路径的最大值。

我们用树链剖分，把路径分成logn个连续的区间，然后对这些区间的补集（即不在路径上的所有点）扔进此条路径的值。线段树的每个节点维护一个可以删除的大根堆。询问时对线段树上所有包含点x的节点的堆顶取个max即可。

时间复杂度$O(nlog^3n)$，空间复杂度$O(nlog^2n)$，bzoj新数据丧心病狂卡内存，不过把求重儿子时>改成>=就可以过了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 200010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(S==T) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=0,fa[N],son[N],sz[N],dep[N],dfn[N],tp[N],dfnum=0;
struct edge{
    int to,next,val;
}data[N<<1];
struct quer{
    int x,y,val;
}qq[N];
struct Lines{
    int l,r;
    friend bool operator<(Lines a,Lines b){return a.l<b.l;}
}a[500];
struct heap{
    priority_queue<int>q1,q2;
    inline void push(int x){q1.push(x);}
    inline void erase(int x){q2.push(x);}
    inline int top(){
        while(!q2.empty()&&q1.top()==q2.top()) q1.pop(),q2.pop();
        if(q1.empty()) return -1;return q1.top();
    }
}tr[N<<2];
inline void dfs1(int x){
    sz[x]=1;son[x]=0;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;
        fa[y]=x;dep[y]=dep[x]+1;dfs1(y);sz[x]+=sz[y];
        if(sz[y]>=sz[son[x]]) son[x]=y;
    }
}
inline void dfs2(int x,int top){
    tp[x]=top;dfn[x]=++dfnum;
    if(son[x]) dfs2(son[x],top);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||y==son[x]) continue;dfs2(y,y);
    }
}
inline void add(int p,int l,int r,int x,int y,int val){
    if(x<=l&&r<=y){tr[p].push(val);return;}
    int mid=l+r>>1;
    if(x<=mid) add(p<<1,l,mid,x,y,val);
    if(y>mid) add(p<<1|1,mid+1,r,x,y,val);
}
inline void del(int p,int l,int r,int x,int y,int val){
    if(x<=l&&r<=y){tr[p].erase(val);return;}
    int mid=l+r>>1;
    if(x<=mid) del(p<<1,l,mid,x,y,val);
    if(y>mid) del(p<<1|1,mid+1,r,x,y,val);
}
inline void doadd(int x,int y,int val){
    int tot=0;
    while(tp[x]!=tp[y]){
        if(dep[tp[x]]<dep[tp[y]]) swap(x,y);
        a[++tot].l=dfn[tp[x]];a[tot].r=dfn[x];x=fa[tp[x]];
    }if(dfn[x]>dfn[y]) swap(x,y);
    a[++tot].l=dfn[x];a[tot].r=dfn[y];sort(a+1,a+tot+1);
    int last=0;
    for(int i=1;i<=tot;++i){
        int l=last+1,r=a[i].l-1;
        if(l<=r) add(1,1,n,l,r,val);last=a[i].r;
    }int l=last+1,r=n;if(l<=r) add(1,1,n,l,r,val);
}
inline void dodel(int x,int y,int val){
    int tot=0;
    while(tp[x]!=tp[y]){
        if(dep[tp[x]]<dep[tp[y]]) swap(x,y);
        a[++tot].l=dfn[tp[x]];a[tot].r=dfn[x];x=fa[tp[x]];
    }if(dfn[x]>dfn[y]) swap(x,y);
    a[++tot].l=dfn[x];a[tot].r=dfn[y];sort(a+1,a+tot+1);
    int last=0;
    for(int i=1;i<=tot;++i){
        int l=last+1,r=a[i].l-1;
        if(l<=r) del(1,1,n,l,r,val);last=a[i].r;
    }int l=last+1,r=n;if(l<=r) del(1,1,n,l,r,val);
}
inline int ask(int p,int l,int r,int x){
    if(l==r) return tr[p].top();
    int mid=l+r>>1,res=tr[p].top();
    if(x<=mid) return max(res,ask(p<<1,l,mid,x));
    else return max(res,ask(p<<1|1,mid+1,r,x));
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs1(1);dfs2(1,1);
    for(int i=1;i<=m;++i){
        int op=read(),x=read();
        if(op==2){printf("%d\n",ask(1,1,n,dfn[x]));continue;}
        if(op==1){dodel(qq[x].x,qq[x].y,qq[x].val);continue;}
        qq[i].x=x;qq[i].y=read();qq[i].val=read();
        doadd(qq[i].x,qq[i].y,qq[i].val);
    }return 0;
}