[NOI2004]郁闷的出纳员——[splay]

【题目描述】

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。

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

$SampleInput$

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

$SampleOutput$

10
20
-1
2

【题意分析】

I操作直接插入x，注意低于初始工资直接return，不然挂
A操作暴力全部节点权值都加上x
S操作的话，先找到权值为x+lim(即调整后正好要退出公司的那个节点)节点的后缀，将它旋到根，那么它的左子树全部是要离开的，注意maintain
F操作直接Kth，因为插了哨兵所以是Kth(x+1)

Code:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#define MAXN 300000
#define INF 1 << 29
using namespace std;

int n, lim, ans, sz;

struct _splay {
    
    int son[MAXN][3], size[MAXN], father[MAXN], cnt[MAXN], val[MAXN], root;
    
    inline void maintain (int x) {
        size[x] = size[son[x][0]] + size[son[x][1]] + cnt[x];
    }
    
    inline void rotate (int x) {
        int y = father[x], z = father[y];
        int k = son[y][1] == x, kk = son[z][1] == y;
        son[z][kk] = x;
        father[x] = z;
        son[y][k] = son[x][k ^ 1];
        father[son[x][k ^ 1]] = y; 
        son[x][k ^ 1] = y;
        father[y] = x;
        maintain (y), maintain (x);
    }
    
    inline void splay (int x, int goal) {
        while (father[x] != goal) {
            int y = father[x], z = father[y];
            if (z != goal)
                (son[y][1] == x) ^ (son[z][1] == y)
                    ? rotate (x) : rotate (y);
            rotate (x);
        }
        if (! goal) root = x;
    }
    
    inline void find (int x) {
        int now = root;
        while (son[now][x > val[now]] && x != val[now])
            now = son[now][x > val[now]];
        splay (now, 0);
    }
    
    inline void addition (int x) {
        for (register int i = 1; i <= sz; i++) val[i] += x;
    }
    
    inline void subtraction (int x) {
        for (register int i = 1; i <= sz; i++) val[i] -= x;
    }
    
    inline int succ (int x) {
        int now = root, nxt;
        while (now) {
            (val[now] >= x)//查询第一个大于等于的
                ? nxt = now, now = son[now][0]
                : now = son[now][1];
        }
        return nxt;
    }
    
    inline void remove (int x) {
        int low = succ (lim + x);
        splay (low, 0), ans += size[son[low][0]];
        son[low][0] = 0, maintain (low);
        subtraction (x);
    }
    
    int Kth (int x, int rank) {//查询的是第rank大
        if (son[x][1] && rank <= size[son[x][1]]) return Kth (son[x][1], rank);
        return (rank <= size[son[x][1]] + cnt[x])
            ? x : Kth (son[x][0], rank - size[son[x][1]] - cnt[x]);
    }
    
    inline void insert (int x) {
        if (x < lim) return;//初始工资低于下限不插入
        int now = root, fa = 0;
        while (val[now] != x && now) fa = now, now = son[now][x > val[now]];
        if (now) cnt[now]++;
        else {
            now = ++sz;
            if (fa) son[fa][x > val[fa]] = now;
            size[now] = cnt[now] = 1;
            son[now][0] = son[now][1] = 0;
            val[now] = x, father[now] = fa;
        }
        splay (now, 0);
    }
    
}tree;

int main () {
    scanf ("%d%d", &n, &lim), tree.insert (INF), tree.insert (-INF);
    for (register int i = 1; i <= n; i++) {
        char opt[5]; int x; scanf ("%s%d", opt, &x);
        if (opt[0] == 'I') tree.insert (x);
        if (opt[0] == 'A') tree.addition (x);
        if (opt[0] == 'S') tree.remove (x);
        if (opt[0] == 'F') {
            if (x >= tree.size[tree.root]) puts ("-1");//如果x比总结点个数还要多，输出-1
            else printf ("%d\n", tree.val[tree.Kth (tree.root, x + 1)]);
        }
    }
    printf ("%d\n", ans);
    return 0;
}