[BZOJ1146][CTSC2008]网络管理Network(二分+树链剖分+线段树套平衡树)

时间:2021-07-14 10:56:46

题意:树上单点修改,询问链上k大值。

思路:

1.DFS序+树状数组套主席树

首先按照套路,关于k大值的问题,肯定要上主席树,每个点维护一棵权值线段树记录它到根的信息。

关于询问,就是Que(u)+Que(v)-Que(LCA(u,v))-Que(father(LCA(u,v))),线段树上二分查询第k大即可。

考虑修改,修改一个点影响到的点为它子树中的所有点,所以要用DFS序。于是这实际上是区间修改单点询问的问题,外面套一层主席树即可。

$O(n\log^2 n)$

2.二分+树链剖分+线段树套平衡树

由于是链上询问,考虑树链剖分。首先二分答案mid,然后问题转化为求mid在这条链上的排名,在树链上用线段树维护,每棵树中维护一棵平衡树,用来求排名。

$O(n\log^4 n)$理论上会超但实际上常数比较小。

第一种码力太弱码不出来,写了第二种,速度OJ上倒数(空间倒是十分优秀)。

 #include<cstdio>
#include<cstring>
#include<algorithm>
#define ls (x<<1)
#define rs (ls|1)
#define lson ls,L,mid
#define rson rs,mid+1,R
#define rep(i,l,r) for (int i=(l); i<=(r); i++)
#define For(i,x) for (int i=h[x],k; i; i=nxt[i])
using namespace std; const int N=,M=;
bool vis[N];
int n,Q,cnt,tot,nd,tmp,tim,ans,u,v;
int val[M],lc[M],rc[M],hp[M],sz[M],w[M];
int T[N],pos[N],b[N<<],fa[N],son[N],dep[N],top[N],root[N<<];
int h[N],to[N<<],nxt[N<<];
struct Que{ int k,a,b; }q[N]; void add(int u,int v){ to[++cnt]=v; nxt[cnt]=h[u]; h[u]=cnt; } void dfs1(int x){
sz[x]=;
For(i,x) if ((k=to[i])!=fa[x]){
fa[k]=x; dep[k]=dep[x]+; dfs1(k); sz[x]+=sz[k];
if (sz[k]>sz[son[x]]) son[x]=k;
}
} void dfs2(int x,int tp){
top[x]=tp; pos[x]=++tim;
if (son[x]) dfs2(son[x],tp);
For(i,x) if ((k=to[i])!=fa[x] && k!=son[x]) dfs2(k,k);
} int lca(int u,int v){
for (; top[u]!=top[v]; u=fa[top[u]])
if (dep[top[u]]<dep[top[v]]) swap(u,v);
return (dep[u]<dep[v]) ? u : v;
} void rrot(int &x){
int y=lc[x]; lc[x]=rc[y]; rc[y]=x;
sz[y]=sz[x]; sz[x]=sz[lc[x]]+sz[rc[x]]+w[x]; x=y;
} void lrot(int &x){
int y=rc[x]; rc[x]=lc[y]; lc[y]=x;
sz[y]=sz[x]; sz[x]=sz[lc[x]]+sz[rc[x]]+w[x]; x=y;
} void ins(int &x,int k){
if (!x) { x=++nd; sz[x]=w[x]=; hp[x]=rand(); val[x]=k; return; }
sz[x]++;
if (k==val[x]) { w[x]++; return; }
if (k<val[x]){ ins(lc[x],k); if (hp[lc[x]]<hp[x]) rrot(x); }
else{ ins(rc[x],k); if (hp[rc[x]]<hp[x]) lrot(x); }
} void del(int &x,int k){
if (val[x]==k){
if (w[x]>) { sz[x]--; w[x]--; return; }
if (!lc[x] || !rc[x]) { x=lc[x]+rc[x]; return; }
if (hp[lc[x]]<hp[rc[x]]) rrot(x),del(x,k);
else lrot(x),del(x,k);
return;
}
sz[x]--;
if (k<val[x]) del(lc[x],k); else del(rc[x],k);
} void getrank(int x,int k){
if (!x) return;
if (val[x]==k) tmp+=sz[rc[x]];
else if (k>val[x]) getrank(rc[x],k); else tmp+=sz[rc[x]]+w[x],getrank(lc[x],k);
} void mdf(int x,int L,int R,int pos,int a,int b){
if (~a) del(root[x],a); ins(root[x],b);
if (L==R) return;
int mid=(L+R)>>;
if (pos<=mid) mdf(lson,pos,a,b); else mdf(rson,pos,a,b);
} void que(int x,int L,int R,int l,int r,int k){
if (L==l && r==R){ getrank(root[x],k); return; }
int mid=(L+R)>>;
if (r<=mid) que(lson,l,r,k);
else if (l>mid) que(rson,l,r,k);
else que(lson,l,mid,k),que(rson,mid+,r,k);
} void Get(int x,int f,int k){
while (top[x]!=top[f]) que(,,n,pos[top[x]],pos[x],k),x=fa[top[x]];
que(,,n,pos[f],pos[x],k);
} void solve(int x,int y,int k){
int t=lca(x,y); ans=-;
if (dep[x]+dep[y]-dep[t]-dep[fa[t]]<k){ puts("invalid request!"); return; }
int L=,R=tot;
while (L<=R){
int mid=(L+R)>>;
tmp=; Get(x,t,mid); Get(y,t,mid);
if (T[t]>mid) tmp--;
if (tmp<k) R=mid-,ans=mid; else L=mid+;
}
printf("%d\n",b[ans]);
} int main(){
freopen("bzoj1146.in","r",stdin);
freopen("bzoj1146.out","w",stdout);
scanf("%d%d",&n,&Q);
rep(i,,n) scanf("%d",&T[i]),b[++tot]=T[i];
rep(i,,n) scanf("%d%d",&u,&v),add(u,v),add(v,u);
dep[]=; dfs1(); dfs2(,);
rep(i,,Q){
scanf("%d%d%d",&q[i].k,&q[i].a,&q[i].b);
if (!q[i].k) b[++tot]=q[i].b;
}
sort(b+,b+tot+); tot=unique(b+,b+tot+)-b-;
rep(i,,n) T[i]=lower_bound(b+,b+tot+,T[i])-b;
rep(i,,Q) if (!q[i].k) q[i].b=lower_bound(b+,b+tot+,q[i].b)-b;
rep(i,,n) mdf(,,n,pos[i],-,T[i]);
rep(i,,Q)
if (!q[i].k) mdf(,,n,pos[q[i].a],T[q[i].a],q[i].b),T[q[i].a]=q[i].b;
else solve(q[i].a,q[i].b,q[i].k);
return ;
}