【洛谷】2464 [SDOI2008]郁闷的小J 平衡树

题目传送门

这题的思路好诡异啊……

最近在看有关平衡树的一些东西，看到书上的课后习题有这题。

看到这题我就傻了——这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;}