[题解]bzoj1503(NOI2004)郁闷的出纳员

Description

OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。好了，现在你已经对我的工作了解不少了。正如你猜的那样
4000
，我想请你编一个工资统计程序。怎么样，不是很困难吧？

Input

Output

输出文件的行数为F命令的条数加一。对于每条F命令，你的程序要输出一行，仅包含一个整数，为当前工资第k多的员工所拿的工资数，如果k大于目前员工的数目，则输出-1。输出文件的最后一行包含一个整数，为离开公司的员工的总数。

Sample Input

9 10

I 60

I 70

S 50

F 2

I 30

S 15

A 5

F 1

F 2

Sample Output

10

20

-1

2

HINT

I命令的条数不超过100000 A命令和S命令的总条数不超过100 F命令的条数不超过100000 每次工资调整的调整量不超过1000 新员工的工资不超过100000

Solution

        这题可以用任意一种平衡树做，注意有相同工资的员工，插入的时候可以每个节点记录一个cnt表示有几个重复的员工。加减工资可以打标记（我就是这么写的），但由于都是统一加减，所以可以记录一个delta工资变化量，就可以不用打标记，但是要注意新插入的员工工资要减去delta来保证所有员工的变化量统一。删除的时候插入一个工资为最小工资的员工，把它旋转到根，删除右子树（为了方便统计第k大，我的平衡树是左子树>根>右子树，当然也可以反过来），记录下删了几个，再把额外插入的这个员工删掉。

       注意！注意！这题最坑的一点是统计走掉的员工数，但是如果是那种插入时就工资低于标准的那种，是不算在内的。这个坑了我一上午。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

inline int read(){
int x=0,f=1;char ch=getchar();
for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
return x*f;
}

const int maxn=100010;
struct node{
int fa,ch[2],size,data,add,cnt;
}T[maxn];
int n,minn,num=0,root=0,tot=0;

void create(int x){
T[++num].data=x;
T[num].cnt=T[num].size=1;
T[num].add=T[num].ch[0]=T[num].ch[1]=0;
}
void pushdown(int x){
if(!T[x].add)return;
if(T[x].ch[0]){
T[T[x].ch[0]].data+=T[x].add;
T[T[x].ch[0]].add+=T[x].add;
}
if(T[x].ch[1]){
T[T[x].ch[1]].data+=T[x].add;
T[T[x].ch[1]].add+=T[x].add;
}
T[x].add=0;
}
void update(int x){
if(!x)return;
T[x].size=T[x].cnt;
if(T[x].ch[0])T[x].size+=T[T[x].ch[0]].size;
if(T[x].ch[1])T[x].size+=T[T[x].ch[1]].size;
}
int getson(int x){
return x==T[T[x].fa].ch[1];
}
void rotate(int p){
if(!T[p].fa)return;
int k=getson(p),fa=T[p].fa;
int fafa=T[fa].fa;
pushdown(fa);pushdown(p);
    T[fa].ch[k]=T[p].ch[k^1];
if(T[p].ch[k^1])T[T[p].ch[k^1]].fa=fa;
T[p].ch[k^1]=fa;
T[fa].fa=p;
T[p].fa=fafa;
if(fafa)T[fafa].ch[fa==T[fafa].ch[1]]=p;
update(fa);update(p);
}
void Splay(int u){
for(int fa;(fa=T[u].fa);rotate(u)){
if(T[fa].fa){
rotate((getson(u)==getson(fa))?fa:u);
}
}
root=u;
}
void Insert(int x){
if(!root){
create(x);
root=num;T[root].fa=0;
return;
}
int p=root,fa=0;
while(p){
pushdown(p);
if(T[p].data==x){
T[p].cnt++;
update(p);update(fa);
Splay(p);return;
}
fa=p;
p=T[p].ch[x<T[p].data];
}
create(x);
T[num].fa=fa;
T[fa].ch[x<T[fa].data]=num;
update(num);update(fa);
Splay(num);
}
void fixup(){
Insert(minn);
tot+=T[T[root].ch[1]].size;
T[root].ch[1]=0;
if(T[root].cnt>1){
T[root].cnt--;
update(root);
}
else{
root=T[root].ch[0];
T[root].fa=0;
}
}
void add(int x){
T[root].add+=x;
T[root].data+=x;
}
void sub(int x){
T[root].add-=x;
T[root].data-=x;
fixup();
}
int find(int x){
int p=root;
while(1){
pushdown(p);
if(T[p].ch[0]&&x<=T[T[p].ch[0]].size)p=T[p].ch[0];
else{
int temp=T[T[p].ch[0]].size+T[p].cnt;
if(x<=temp)return T[p].data;
x-=temp;
p=T[p].ch[1];
}
}
}
void Init(){
n=read();minn=read();
root=0;
}
void Work(){
char opt;int k;
while(n--){
opt=getchar();
while(opt==' '||opt=='\n')opt=getchar();
k=read();
if(opt=='I'){
if(k>=minn)Insert(k);
}
else if(opt=='A')add(k);
else if(opt=='S')sub(k);
else{
if(k>T[root].size)printf("-1\n");
else printf("%d\n",find(k));
}
}
printf("%d\n",tot);
}

int main(){
Init();
Work();
return 0;
}