BZOJ1503：[NOI2004]郁闷的出纳员——题解

http://www.lydsy.com/JudgeOnline/problem.php?id=1503

（题面复制的洛谷的，因为洛谷好看）

题目描述

OIER公司是一家大型专业化软件公司，有着数以万计的员工。作为一名出纳员，我的任务之一便是统计每位员工的工资。这本来是一份不错的工作，但是令人郁闷的是，我们的老板反复无常，经常调整员工的工资。如果他心情好，就可能把每位员工的工资加上一个相同的量。反之，如果心情不好，就可能把他们的工资扣除一个相同的量。我真不知道除了调工资他还做什么其它事情。

工资的频繁调整很让员工反感，尤其是集体扣除工资的时候，一旦某位员工发现自己的工资已经低于了合同规定的工资下界，他就会立刻气愤地离开公司，并且再也不会回来了。每位员工的工资下界都是统一规定的。每当一个人离开公司，我就要从电脑中把他的工资档案删去，同样，每当公司招聘了一位新员工，我就得为他新建一个工资档案。

老板经常到我这边来询问工资情况，他并不问具体某位员工的工资情况，而是问现在工资第k多的员工拿多少工资。每当这时，我就不得不对数万个员工进行一次漫长的排序，然后告诉他答案。

好了，现在你已经对我的工作了解不少了。正如你猜的那样，我想请你编一个工资统计程序。怎么样，不是很困难吧？

如果某个员工的初始工资低于最低工资标准,那么将不计入最后的答案内

输入输出格式

输入格式：

第一行有两个非负整数n和min。n表示下面有多少条命令，min表示工资下界。

接下来的n行，每行表示一条命令。命令可以是以下四种之一：

名称 格式 作用

I命令 I_k 新建一个工资档案，初始工资为k。如果某员工的初始工资低于工资下界，他将立刻离开公司。

A命令 A_k 把每位员工的工资加上k

S命令 S_k 把每位员工的工资扣除k

F命令 F_k 查询第k多的工资

_（下划线）表示一个空格，I命令、A命令、S命令中的k是一个非负整数，F命令中的k是一个正整数。

在初始时，可以认为公司里一个员工也没有。

输出格式：

输出文件的行数为F命令的条数加一。

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

输出文件的最后一行包含一个整数，为离开公司的员工的总数。

输入输出样例

输入样例#1： 

9 10
I 60
I 70
S 50
F 2
I 30
S 15
A 5
F 1
F 2

输出样例#1： 

10
20
-1
2

说明

I命令的条数不超过100000

A命令和S命令的总条数不超过100

F命令的条数不超过100000

每次工资调整的调整量不超过1000

新员工的工资不超过100000

————————————————————————————

维护一棵平衡树，记录一个det记录这棵树上的点最后需要加上det才为真正的值。

于是我们每次插入的人的钱都是k-det。

当我们查询哪些人要被删的时候，我们插入minn-det，然后删掉该点的左子树即可。

那么询问第k大就暴力在平衡树搜即可（方法基本同线段树的搜索）

#include<cstdio>
#include<queue>
#include<cctype>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
const int N=100001;
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
int fa[N],tr[N][2],key[N],cnt[N],size[N];
int root,sz;
inline void clear(int x){
    fa[x]=tr[x][0]=tr[x][1]=key[x]=cnt[x]=size[x]=0;
    return;
}
inline bool get(int x){
    return tr[fa[x]][1]==x;
}
inline void update(int x){
    if(x){  
    size[x]=cnt[x];  
    if(tr[x][0])size[x]+=size[tr[x][0]];  
    if(tr[x][1])size[x]+=size[tr[x][1]];  
    }  
    return;
}
inline void rotate(int x){
    int old=fa[x],oldf=fa[old],which=get(x);
    tr[old][which]=tr[x][which^1];fa[tr[old][which]]=old;  
    fa[old]=x;tr[x][which^1]=old;fa[x]=oldf;
    if(oldf)tr[oldf][tr[oldf][1]==old]=x;
    update(old);update(x);
    return;
}
inline void splay(int x){
    int f=fa[x];
    while(f){
    if(fa[f])rotate((get(x)==get(f)?f:x));
    rotate(x);f=fa[x];
    }
    root=x;
    return;
}
inline void insert(int v){
    if(!root){
    sz++;tr[sz][0]=tr[sz][1]=fa[sz]=0;
    key[sz]=v;cnt[sz]=size[sz]=1;root=sz;
    return;
    }
    int now=root,f=0;
    while(233){
    if(key[now]==v){
        cnt[now]++;update(now);update(f);splay(now);
        break;
    }
    f=now;
    now=tr[now][key[now]<v];
    if(!now){
        sz++;tr[sz][0]=tr[sz][1]=0;fa[sz]=f;
        key[sz]=v;cnt[sz]=size[sz]=1;
        tr[f][key[f]<v]=sz;
        update(f);splay(sz);
        break;
    }
    }
    return;
}
inline void del(){
    if(cnt[root]>1){
    cnt[root]--;
    size[root]--;
    return;
    }
    if(!tr[root][1]){
    clear(root);root=0;return;
    }
    int oldroot=root;root=tr[root][1];fa[root]=0;clear(oldroot);
    return;
}
int query(int k,int x){
    if(size[tr[x][1]]<k&&k<=cnt[x]+size[tr[x][1]])return key[x];
    else if(size[tr[x][1]]<k)return query(k-size[tr[x][1]]-cnt[x],tr[x][0]);
    else return query(k,tr[x][1]);
}
int main(){
    int n=read();
    int minn=read();
    int det=0;
    int delnum=0;
    for(int i=1;i<=n;i++){
    char ch=0;
    while(ch<'A'||ch>'Z')ch=getchar();
    int k=read();
    if(ch=='I')if(k>=minn)insert(k-det);
    if(ch=='A')det+=k;
    if(ch=='S'){
        det-=k;
        insert(minn-det);
        size[root]-=size[tr[root][0]];
        delnum+=size[tr[root][0]];
        clear(tr[root][0]);
        tr[root][0]=0;
        del();
    }
    if(ch=='F'){
        if(size[root]<k)puts("-1");
        else{
        printf("%d
",query(k,root)+det);
        }
    }
    }
    printf("%d
",delnum);
    return 0;
}