bzoj 4538: [Hnoi2016]网络
查询的是不包含x的链的权值最大值。
树链剖分后，每条链变成O(logn)个区间，那么未被这个链包含的也有logn个区间，在这logn个区间上做修改即可。
因为有删除操作，每个线段树节点开一个堆维护。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#define ll long long
#define inf 1e9
#define eps 1e-8
#define md
#define N 100010
using namespace std;
struct QQ { int l,r;} q[N];
struct yts { int x,t,ne;} e[2*N];
int v[N],dep[N],fa[N][20],val[2*N],sz[N],son[N],pos[N],top[N];
int dfs_cnt,Ans,num,n;
bool ed[2*N];
struct cmp
{
bool operator () (int a,int b) { return val[a]<val[b];}
};
priority_queue<int,vector<int>,cmp> pri[4*N];
bool cmp (QQ a,QQ b) { return a.l<b.l;}
void put(int x,int y)
{
num++; e[num].x=x; e[num].t=y;
e[num].ne=v[x]; v[x]=num;
}
void dfs1(int x)
{
sz[x]=1; son[x]=0;
for (int i=v[x];i;i=e[i].ne)
{
int y=e[i].t;
if (y!=fa[x][0])
{
dep[y]=dep[x]+1; fa[y][0]=x;
dfs1(y);
sz[x]+=sz[y];
if (sz[y]>sz[son[x]]) son[x]=y;
}
}
}
void dfs2(int x,int f)
{
pos[x]=++dfs_cnt; top[x]=f;
if (son[x]) dfs2(son[x],f);
for (int i=v[x];i;i=e[i].ne)
{
int y=e[i].t;
if (y!=fa[x][0]&&y!=son[x]) dfs2(y,y);
}
}
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 insert(int i,int l,int r,int ql,int qr,int d)
{
if (ql<=l&&r<=qr) { pri[i].push(d); return;}
int mid=(l+r)>>1;
if (ql<=mid) insert(i<<1,l,mid,ql,qr,d);
if (mid+1<=qr) insert(i<<1|1,mid+1,r,ql,qr,d);
}
void query(int i,int l,int r,int x)
{
while (!pri[i].empty())
{
int d=pri[i].top();
if (ed[d]) pri[i].pop();
else
{
Ans=max(Ans,val[d]);
break;
}
}
if (l==r) return;
int mid=(l+r)>>1;
if (x<=mid) query(i<<1,l,mid,x);
else query(i<<1|1,mid+1,r,x);
}
int main()
{
int m;
scanf("%d%d",&n,&m);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
put(x,y); put(y,x);
}
dep[1]=1; dfs1(1); dfs2(1,1); build_lca();
for (int i=1;i<=m;i++)
{
int ty;
scanf("%d",&ty);
if (ty==0)
{
int x,y;
scanf("%d%d%d",&x,&y,&val[i]);
int w=0,z=lca(x,y);
while (dep[top[x]]>dep[z])
{
q[++w]=(QQ){pos[top[x]],pos[x]};
x=fa[top[x]][0];
}
while (dep[top[y]]>dep[z])
{
q[++w]=(QQ){pos[top[y]],pos[y]};
y=fa[top[y]][0];
}
if (x!=z) q[++w]=(QQ){pos[z],pos[x]};
else if (y!=z) q[++w]=(QQ){pos[z],pos[y]};
else q[++w]=(QQ){pos[z],pos[z]};
q[++w]=(QQ){n+1,n+1};
sort(q+1,q+w+1,cmp);
int top=0;
for (int j=1;j<=w;j++)
if (q[j-1].r<q[j].l-1) insert(1,1,n,q[j-1].r+1,q[j].l-1,i);
}
else if (ty==1)
{
int x;
scanf("%d",&x);
ed[x]=1;
}
else
{
int x;
scanf("%d",&x);
Ans=-1; query(1,1,n,pos[x]);
printf("%d\n",Ans);
}
}
return 0;
}