《郁闷的出纳员》此题曾一度令人十分郁闷,原因是全局标记的转化容易出错。
可以将数值视作一维系量,则有
r=r0+r′
,其中
r
为真实工资,
// 平衡树中记录相对工资,实际为绝对工资,a0为系工资
// 绝对工资=系工资+相对工资
// 增加全部工资时,实际上是增加了绝对工资,相对工资不变,因此系工资增加
// 获取工资时,获取的是绝对工资,有绝对工资=系工资+相对工资
// 放入工资时,放入的是绝对工资,有相对工资=绝对工资-系工资
// low中记录的是最低的相对工资,最低绝对工资=系工资+最低相对工资
下面给出参考代码。
#include<cstdio>
using namespace std;
int fa[100005],ch[100005][2],k[100005],cnt[100005],size[100005],m,ind=0,root=0,t1,t2,ans=0,n,low,a0=0;
void update(int p){ch[0][0]=ch[0][1]=size[0]=0;size[p]=size[ch[p][0]]+size[ch[p][1]]+cnt[p];}
void rotate(int p){int q=fa[p],y=fa[q],x=ch[q][1]==p;ch[q][x]=ch[p][x^1];fa[ch[q][x]]=q;ch[p][x^1]=q;fa[q]=p;fa[p]=y;if(y)if(ch[y][0]==q)ch[y][0]=p;else if(ch[y][1]==q)ch[y][1]=p;update(q);update(p);}
void splay(int x){for(int y;y=fa[x];rotate(x))if(fa[y])rotate((x==ch[y][0])==(y==ch[fa[y]][0])?y:x);root=x;}
int search(int v){int p=root;for(;p&&k[p]!=v;p=ch[p][k[p]<v]);if(k[p]==v)return p;else return 0;}
void insert(int v){if(!root)root=++ind,k[root]=v,cnt[root]=size[root]=1,fa[root]=ch[root][0]=ch[root][1]=0;else if(search(v))splay(search(v)),cnt[root]++,size[root]++;else{int x=root,y;for(;y=ch[x][k[x]<v];x=y);y=++ind;k[y]=v;cnt[y]=size[y]=1;fa[y]=x;ch[y][0]=ch[y][1]=0;ch[x][k[x]<v]=y;splay(y);}}
int kth(int _k){int p=root,k=_k;for(;p;)if(ch[p][0]){if(k>size[ch[p][0]]&&k<=size[ch[p][0]]+cnt[p])return p;if(k<=size[ch[p][0]])p=ch[p][0];else k-=(size[ch[p][0]]+cnt[p]),p=ch[p][1];}else if(k<=cnt[p])return p;else k-=cnt[p],p=ch[p][1];return 0;}
void refresh(int rf){
insert(rf);
splay(search(rf));
ans+=size[ch[root][0]]+cnt[root]-1;
root=ch[root][1];
fa[root]=0;
}
int main(){
scanf("%d%d",&n,&low);
for(int i=0;i<n;i++){
int t;char c[10];
scanf("%s%d",c,&t);
switch(c[0]){
case 'I':if(t>=low)insert(t-a0);break;
case 'A':a0+=t;refresh(low-a0-1);break;
case 'S':a0-=t;refresh(low-a0-1);break;
case 'F':printf(" %d\n",(t>size[root])?-1:k[kth(size[root]-t+1)]+a0);break;
}
}
printf(" %d\n",ans);
}