NOI2004郁闷的出纳员

输入描述 Input Description

第一行有两个非负整数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是一个正整数。

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

输出描述 Output Description

输出文件的行数为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

经典splay。参考黄学长代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int sum[100001],t[100001],fa[100001];
int ts[100001][2],m,size=0,root,delta,l;
int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l,r;
    if (ts[y][0]==x)l=0;else l=1;r=l^1;
    if (y==k) k=x;
    else if (ts[z][0]==y)ts[z][0]=x;else ts[z][1]=x;
    fa[x]=z;fa[y]=x;
    ts[y][l]=ts[x][r];
    fa[ts[x][r]]=y;
    ts[x][r]=y;
    sum[x]+=1+sum[ts[y][r]];
    sum[y]=sum[y]-1-sum[ts[x][l]];
}

void splay(int x,int &k){
    int y,z;
    while (x!=k){
        y=fa[x];z=fa[y];
        if (y!=k){
            if (ts[z][0]==y^ts[y][0]==x)rotate(x,k);else rotate(y,k);
        }
        rotate(x,k);
    }
}

void ins(int &k,int x,int pre){
    if (k==0){
        size++;k=size;
        t[k]=x;sum[k]=1;fa[k]=pre;
        splay(k,root);
        return;
    }
    sum[k]++;
    if (x>=t[k]) ins(ts[k][1],x,k);else ins(ts[k][0],x,k);
}

int find(int k,int x){
    if (x==sum[ts[k][1]]+1) return t[k];
    if (sum[ts[k][1]]>=x) return find(ts[k][1],x);
    else return find(ts[k][0],x-1-sum[ts[k][1]]);    
}

int dec(int &x,int pre){
    if(x==0)return 0;
    int k;
    if(t[x]+delta<m){
        k=dec(ts[x][1],x)+sum[ts[x][0]]+1;
        sum[ts[x][1]]=sum[x]-k;
        x=ts[x][1];fa[x]=pre;
    }
    else{
        k=dec(ts[x][0],x);
        sum[x]-=k;
    }
    return k;
}

int main(){
    int n=read();m=read();
    for (int i=1;i<=n;i++){
        char s[2];
        scanf("%s",s);
        int k=read();
        if (s[0]=='I'&&k>=m)ins(root,k-delta,0);
        if (s[0]=='A')delta+=k;
        if (s[0]=='S'){
            delta-=k;
            l+=dec(root,0);
        }
        if (s[0]=='F'){
            printf("%d\n",k<=sum[root]?find(root,k)+delta:-1);
        }
    }
    printf("%d",l);
    return 0;
}