bzoj 1503[NOI 2004] 郁闷的出纳员

题目大意：

给4种操作

I：添加一个员工工资信息

A：增加所有员工的工资

S：减少所有员工的工资

F：询问工资第k高的员工的工资情况

自己做的第一道splay树的题目，初学找找感觉

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
int n,m,w,limit;
const int N = 100010;
#define ls ch[x][0]
#define rs ch[x][1]
struct SplayTree{
    //sum[i]记录i以及其子树中的点的总个数,cnt[i]记录与i号位置取值相等的点的个数
    int val[N] , cnt[N] , sum[N];
    int all; //统计离开公司的员工的总人数，也就是相当于计算删除的点的个数
    int ch[N][2];
    int pre[N];
    int rt , top;

    void init()
    {
        ch[0][0] = ch[0][1] = pre[0] = sum[0] = cnt[0] = 0;
        all = rt = top = 0;
    }

    void newNode(int &x , int c)
    {
        x = ++top;
        ch[x][0] = ch[x][1] = pre[x] = 0;
        cnt[x] = sum[x] = 1 , val[x]=c;
    }
    //通过左右子节点更新父节点
    void up(int x){
        sum[x] = sum[ch[x][0]]+sum[ch[x][1]]+cnt[x];
    }

    void Rotate(int x , int f) //f==1表示右旋，也就是x属于父亲的左子树上
    {
        int y=pre[x];
        ch[y][!f] = ch[x][f];
        pre[ch[x][f]]=y;
        if(pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;
        pre[x]=pre[y];
        pre[y]=x;
        ch[x][f]=y;
        up(y);
    }

    void Splay(int x , int goal)
    {
        while(pre[x] != goal){
            if(pre[pre[x]] == goal) Rotate(x , ch[pre[x]][0] == x);
            else{
                int y=pre[x] , z=pre[y];
                int f=(ch[z][0] == y);
                if(ch[y][f] == x) Rotate(x , !f) , Rotate(x , f);
                else Rotate(y,f) , Rotate(x,f);
            }
        }
        up(x);
        if(goal == 0) rt = x;
    }

    void insert(int &x , int key , int fa)
    {
        //一直向下找到空的叶子节点插入当前的值
        if(!x){
            newNode(x , key);
            pre[x]=fa;
            Splay(x , 0);
            return;
        }
        if(key == val[x]){
            cnt[x]++;
            sum[x]++;
            Splay(x,0);
            return ;
        }
        //点插入左子树
        else if(key<val[x]){
            insert(ch[x][0] , key , x);
        }
        //点插入右子树
        else {
            insert(ch[x][1] , key , x);
        }
        up(x);
    }

    void del(int &x , int fa)
    {
        //一直访问到空的叶子节点结束
        if(!x) return ;
        //当前点的工资满足要求，那么只要考虑其左子树上要删除多少点
        if(val[x] >= limit) del(ch[x][0] , x);
        else{
            /*当前点的工资不满足要求，那么这个点和其左子树都是不满足要求的
            ,all记录当前点和左子树删除的点的总数*/
            all+=sum[ch[x][0]]+cnt[x];
            x=ch[x][1];
            //当前点被删除，连接关系要进行修改
            pre[x]=fa;
            if(fa == 0) rt = x;
            del(x,fa);
        }
        if(x) up(x);
    }

    void update()
    {
        del(rt , 0);
    }

    int find_kth(int x , int k)
    {
        if(k<sum[ch[x][0]]+1) return find_kth(ch[x][0] , k);
        else if(k > sum[ch[x][0]]+cnt[x])
            return find_kth(ch[x][1] , k-sum[ch[x][0]]-cnt[x]);
        else{
            Splay(x , 0);
            return x;
        }
    }
}spt;

int main()
{
   // freopen("a.in" , "r" , stdin);
    char op[10];
    int t;
    while(~scanf("%d%d" , &n , &m))
    {
        spt.init();
        for(int i=0 ; i<n ; i++){
            scanf("%s%d" , op , &t);
            if(op[0] == 'I'){
                if(t<m)
                    continue;
                spt.insert(spt.rt , t-w , 0);
            }
            else if(op[0] == 'A') w+=t;
            else if(op[0] == 'S'){
                w-=t;
                limit=m-w;
                spt.update();
            }
            else{
                int sum = spt.sum[spt.rt];
                if(t>sum) printf("-1
");
                else{
                    printf("%d
" , spt.val[spt.find_kth(spt.rt , sum-t+1)]+w);
                }
            }
          //  cout<<"sum: "<<spt.sum[spt.rt]<<endl;
        }
        printf("%d
" , spt.all);
    }
    return 0;
}