这题的思路好诡异啊……
最近在看有关平衡树的一些东西,看到书上的课后习题有这题。
看到这题我就傻了——这TMD怎么用平衡树做?
题解给出的思路是对每本书的编号离散化后建立平衡树森林,权值为这本书所在的位置。
对于每个修改就把这本书从原来的平衡树中删掉,然后插入新的平衡树中。
询问非常的简单,差分一下就行了。
(话说这题如果用分块做感觉要好想多了……)
附上AC代码:
#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;
const int N=2e5+10;
struct query{
int x,y,w;
}q[N];
int n,m,a[N],s[N],num,len,bh[N],t,sy[N],rt[N],f[N],w[N],sz[N],size,ch[N][2];
char c;
inline char nc(void){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
a*=f;return;
}
inline int find(int x){
int l=1,r=len,mid,ret=0;
while (l<=r)
if (bh[(mid=(l+r)>>1)]<=x) ret=mid,l=mid+1;
else r=mid-1;
return ret;
}
inline void updata(int x){sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;}
inline void rotate(int x){
int y=f[x],op=(ch[y][1]==x);
ch[y][op]=ch[x][op^1];
if (ch[x][op^1]) f[ch[x][op^1]]=y;
f[x]=f[y];
if (f[y]) ch[f[y]][ch[f[y]][1]==y]=x;
f[y]=x,ch[x][op^1]=y;
return updata(y),updata(x);
}
inline void splay(int x,int ed,int wz){
for (int fa=f[x]; fa!=ed; rotate(x),fa=f[x])
if (f[fa]!=ed) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
if (!ed) rt[wz]=x;
return;
}
inline void ist(int wz,int x){
int now=rt[wz];
while (ch[now][w[now]<x]) now=ch[now][w[now]<x];
w[++size]=x,f[size]=now,ch[size][0]=ch[size][1]=0;
if (now) ch[now][w[now]<x]=size;
return splay(size,0,wz);
}
inline void del(int wz,int x){
int now=rt[wz];
while (w[now]!=x) now=ch[now][w[now]<x];
splay(now,0,wz);
if (!ch[rt[wz]][0]) {f[rt[wz]=ch[rt[wz]][1]]=0; return;}
now=ch[rt[wz]][0];
while (ch[now][1]) now=ch[now][1];
splay(now,rt[wz],wz);
f[ch[now][1]=ch[rt[wz]][1]]=now,f[rt[wz]=now]=0;
return;
}
inline int ask(int wz,int x){
int now=rt[wz],ans=0;
while (now)
if (w[now]>x) ans=now,now=ch[now][0];
else now=ch[now][1];
splay(ans,0,wz);
return sz[ch[rt[wz]][0]];
}
int main(void){
read(n),read(m);
for (int i=1; i<=n; ++i) read(a[i]),s[i]=a[i];
num=n;
for (int i=1; i<=m; ++i){
c=nc();
while (c!='Q'&&c!='C') c=nc();
read(q[i].x),read(q[i].y);
switch (c){
case 'C': s[++num]=q[i].y; break;
case 'Q': read(q[i].w); break;
}
}
sort(s+1,s+1+num);
for (int i=1; i<=num; ++i) if (s[i]!=s[i+1]) bh[++len]=s[i];
for (int i=1; i<=len; ++i) ist(i,2e9);
for (int i=1; i<=n; ++i){
sy[i]=(t=find(a[i]));
ist(t,i);
}
for (int i=1; i<=m; ++i)
if (q[i].w){
t=find(q[i].w);
printf("%d\n",ask(t,q[i].y)-ask(t,q[i].x-1));
}
else{
t=find(q[i].y);
del(sy[q[i].x],q[i].x);
sy[q[i].x]=t;
ist(t,q[i].x);
}
return 0;
}