题目描述:
n个点的树,m次操作,每次可以加入一条权值为wi的路径(不改变树的结构,只用于询问),删除之前加入的一条路径,或者询问不经过某个点u的路径的最大权值。
n<=100000,m<=200000
题目分析:
把权值在dfs序上赋给路径树链剖分之外的点,我们得到:
比较暴力的树链剖分+线段树+堆的nlog3n做法
不经过某个点,可以转化为权值>=mid的路径是否都经过这个点,问题变为二分+树上路径的交,我们得到:
整体二分+树上差分树状数组统计经过路径条数的nlog2n做法
然而树链的交可以O(1)求,二分也可以直接在线段树上二分,于是我们得到:
欧拉序O(nlogn)~O(1)求LCA+O(1)求树链的交+线段树二分的nlogn做法
顺便%Freopen的blog
描述一下求树链的交的过程,设两条路径为 ( a , b ) , ( u , v ) (a,b),(u,v) (a,b),(u,v)
- 求 ( a , u ) , ( a , v ) , ( b , u ) , ( b , v ) (a,u),(a,v),(b,u),(b,v) (a,u),(a,v),(b,u),(b,v)的LCA,取其中深度最大的两个记为 p , q p,q p,q(dep[p]>=dep[q])
- 如果 p ≠ q p\neq q p=q,则路径交为 ( p , q ) (p,q) (p,q)
- 如果 p = = q p==q p==q,则判断 L C A ( a , b ) = = p ∣ ∣ L C A ( u , v ) = = p LCA(a,b)==p||LCA(u,v)==p LCA(a,b)==p∣∣LCA(u,v)==p,如果成立则交为一个点 ( p , p ) (p,p) (p,p),否则没有交。
稍微画个图理解一下,我也不怎么会严谨证明(懒得想了 ):
至于怎么判断一个点是否在一条链上可以看代码中的Onchain函数,这里就不多说了。
Code(nlogn,求树链交时把0看做没有路径,-1看做路径交为空):
#include<bits/stdc++.h>
#define maxn 200005
using namespace std;
char cb[1<<18],*cs,*ct,wb[2000005],*wt=wb;
#define getc() (cs==ct&&(ct=(cs=cb)+fread(cb,1,1<<18,stdin),cs==ct)?0:*cs++)
inline void read(int &a){
char c;while(!isdigit(c=getc()));
for(a=c-'0';isdigit(c=getc());a=a*10+c-'0');
}
inline void write(int x){
int q[15],len=0;
if(x<0) *wt++='-',x=-x;
while(x) q[++len]=x%10,x/=10;
if(len) while(len) *wt++=q[len--]+'0'; else *wt++='0';
}
const int Log = 17;
int n,m,dep[maxn],st[maxn][Log+1],dfn[maxn],tim,pos[maxn],id[maxn],cnt,lg[maxn];
int fir[maxn],nxt[maxn],to[maxn],tot;
struct Q{
int op,x,y,v;
}q[maxn];
inline void line(int x,int y){nxt[++tot]=fir[x],fir[x]=tot,to[tot]=y;}
void dfs(int u,int ff){
dep[u]=dep[ff]+1,st[dfn[u]=++tim][0]=u;
for(int i=fir[u];i;i=nxt[i]) if(to[i]!=ff) dfs(to[i],u),st[++tim][0]=u;
}
bool cmpv(int i,int j){return q[i].v<q[j].v;}
bool cmpd(int i,int j){return dep[i]>dep[j];}
int t[maxn<<2][2];
int LCA(int u,int v){
if(dfn[u]>dfn[v]) swap(u,v);
int k=lg[(v=dfn[v])-(u=dfn[u])+1];
return dep[u=st[u][k]]<dep[v=st[v-(1<<k)+1][k]]?u:v;
}
bool Onchain(int *a,int x){
if(a[0]==-1) return 0;
if(!a[0]) return 1;
if(dep[x]<dep[LCA(a[0],a[1])]) return 0;
return LCA(a[0],x)==x||LCA(a[1],x)==x;
}
void merge(int *c,int *a,int *b){
if(a[0]==-1||b[0]==-1) {c[0]=c[1]=-1;return;}
if(!a[0]||!b[0]) {c[0]=a[0]+b[0],c[1]=a[1]+b[1];return;}
int p[4]={LCA(a[0],b[0]),LCA(a[0],b[1]),LCA(a[1],b[0]),LCA(a[1],b[1])};
sort(p,p+4,cmpd);
if(p[0]!=p[1]) c[0]=p[0],c[1]=p[1];
else if(LCA(a[0],a[1])==p[0]||LCA(b[0],b[1])==p[0]) c[0]=c[1]=p[0];
else c[0]=c[1]=-1;
}
void insert(int i,int l,int r,int x){
if(l==r){
if(!t[i][0]) t[i][0]=q[id[x]].x,t[i][1]=q[id[x]].y;
else t[i][0]=t[i][1]=0;
return;
}
int mid=(l+r)>>1;
if(x<=mid) insert(i<<1,l,mid,x);
else insert(i<<1|1,mid+1,r,x);
merge(t[i],t[i<<1],t[i<<1|1]);
}
int query(int i,int l,int r,int x){
if(l==r) return q[id[l]].v;
int mid=(l+r)>>1;
if(!Onchain(t[i<<1|1],x)) return query(i<<1|1,mid+1,r,x);
else return query(i<<1,l,mid,x);
}
int main()
{
read(n),read(m);
for(int i=1,x,y;i<n;i++) read(x),read(y),line(x,y),line(y,x);
dfs(1,0);
for(int i=2;i<=tim;i++) lg[i]=lg[i>>1]+1;
for(int j=1;j<=Log;j++)
for(int i=tim-(1<<j)+1;i>=1;i--)
st[i][j]=(dep[st[i][j-1]]<dep[st[i+(1<<(j-1))][j-1]]?st[i][j-1]:st[i+(1<<(j-1))][j-1]);
for(int i=1;i<=m;i++){
read(q[i].op);
if(!q[i].op) read(q[i].x),read(q[i].y),read(q[i].v),id[++cnt]=i;
if(q[i].op==1) read(q[i].v);
if(q[i].op==2) read(q[i].v);
}
sort(id+1,id+1+cnt,cmpv);
for(int i=1;i<=m;i++) pos[id[i]]=i;
for(int i=1;i<=m;i++){
if(!q[i].op) insert(1,1,cnt,pos[i]);
if(q[i].op==1) insert(1,1,cnt,pos[q[i].v]);
if(q[i].op==2) write(Onchain(t[1],q[i].v)?-1:query(1,1,cnt,q[i].v)),*wt++='\n';
}
fwrite(wb,wt-wb,1,stdout);
}