【bzoj4538】[Hnoi2016]网络

我们考虑树剖，线段树上维护一个堆，保存不经过该段区间的路径的权值。 
对于一条路径我们将对于线段树中的区间提取出来，在对于线段树中进行修改。也就是在堆中插入或删除。 
对于一次询问，只要找到包含该点的线段中堆顶权值最大的就行了。

#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
  
typedef long long LL;
  
#define N 200010
  
struct data
{
    int l,r;
}q[N];
  
struct edge
{
    int s,to,next;
}e[N<<1];
int head[N<<1];
int cnt;
  
int dep[N],siz[N],son[N],top[N],pos[N];
int fa[N][20],val[N<<2];
  
bool vis[N];
  
int n,m;
int ans,res,num;
  
int askd;
  
struct cmp
{
    bool operator() (int a,int b) 
    {
        return val[a]<val[b]; 
    }
};
  
priority_queue<int,vector<int>,cmp> Q[N<<2];
  
bool cmp(data a,data b)
{
    return a.l<b.l;
}
  
void link(int x,int y)
{
    e[++cnt]=(edge){x,y,head[x]};
    head[x]=cnt;
}
  
void dfs(int x)
{
    siz[x]=1;
    son[x]=0;
    for (int i=head[x];i;i=e[i].next)
    {
        int t=e[i].to;
        if (t!=fa[x][0])
        {
            dep[t]=dep[x]+1;
            fa[t][0]=x;
            dfs(t);
            siz[x]+=siz[t];
            if (siz[t]>siz[son[x]])
                son[x]=t;
        }
    }
}
  
void dfs2(int x,int cha)
{
    pos[x]=++res;
    top[x]=cha;
    if (son[x])
        dfs2(son[x],cha);
    for (int i=head[x];i;i=e[i].next)
    {
        int t=e[i].to;
        if (t!=fa[x][0] && t!=son[x])
            dfs2(t,t);
    }
}
  
void build_lca()
{
    for (int j=1;j<=18;j++)
        for (int i=1;i<=n;i++)
            if (fa[i][j-1])
                fa[i][j]=fa[fa[i][j-1]][j-1];
}
  
int lca(int x,int y)
{
    if (dep[x]<dep[y])
        swap(x,y);
    int t=dep[x]-dep[y];
    for (int i=18;i>=0;i--)
        if (t & (1<<i))
            x=fa[x][i];
    if (x==y)
        return x;
    for (int i=18;i>=0;i--)
        if (fa[x][i]!=fa[y][i])
            x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
  
void add(int nowl,int nowr,int now,int l,int r,int d)
{
    if (l<=nowl && nowr<=r)
    {
        Q[now].push(d);
        return ;
    }
    int mid=(nowl+nowr)>>1;
    if (l<=mid)
        add(nowl,mid,now<<1,l,r,d);
    if (r>=mid+1)
        add(mid+1,nowr,now<<1|1,l,r,d);
}
  
void query(int nowl,int nowr,int now,int d)
{
    while (!Q[now].empty())
    {
        int k=Q[now].top();
        if (vis[k])
            Q[now].pop();
        else
        {
            num=max(num,val[k]);
            break;
        }
    }
    if (nowl==nowr)
        return ;
    int mid=(nowl+nowr)>>1;
    if (d<=mid)
        query(nowl,mid,now<<1,d);
    else
        query(mid+1,nowr,now<<1|1,d);
}
int main()
{
    scanf("%d%d",&n,&m);
    int x,y;
    for (int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        link(x,y);
        link(y,x);
    }
    dep[1]=1;
    dfs(1);
    dfs2(1,1);
    build_lca();
    for (int i=1;i<=m;i++)
    {
        scanf("%d",&askd);
        if (askd==0)
        {
            scanf("%d%d%d",&x,&y,&val[i]);
            int w=0,z=lca(x,y);
            while (dep[top[x]]>dep[z])
            {
                q[++w]=(data){pos[top[x]],pos[x]};
                x=fa[top[x]][0];
            }
            while (dep[top[y]]>dep[z])
            {
                q[++w]=(data){pos[top[y]],pos[y]};
                y=fa[top[y]][0];
            }
            if (x!=z)
                q[++w]=(data){pos[z],pos[x]};
            else if (y!=z)
                q[++w]=(data){pos[z],pos[y]};
            else
                q[++w]=(data){pos[z],pos[z]};
            q[++w]=(data){n+1,n+1};
            sort(q+1,q+w+1,cmp);
            for (int j=1;j<=w;j++)
                if (q[j-1].r<q[j].l-1)
                    add(1,n,1,q[j-1].r+1,q[j].l-1,i);
        }
        else if (askd==1)
        {
            scanf("%d",&x);
            vis[x]=1;
        }
        else
        {
            scanf("%d",&x);
            num=-1;
            query(1,n,1,pos[x]);
            printf("%d
",num);
        }
    }
    return 0;
}