BZOJ1503【NOI2004】郁闷的出纳员 < Splay >

Problem

【NOI2004】郁闷的出纳员

Description

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

Input

第一行有两个非负整数和。表示下面有多少条命令，表示工资下界。
接下来的行，每行表示一条命令。命令可以是以下四种之一：

名称	格式	作用
命令		新建一个工资档案，初始工资为
命令		把每位员工的工资加上
命令		把每位员工的工资扣除
命令		查询第多的工资

命令、命令、命令中的是一个非负整数，命令中的是一个正整数。
在初始时，可以认为公司里一个员工也没有。
命令的条数不超过
命令和命令的总条数不超过
命令的条数不超过
每次工资调整的调整量不超过
新员工的工资不超过

Output

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

Sample Input

Sample Output

标签：Splay

Solution

基础题。

由于加和减都是作用于全局，我们可以单独记当前加和减操作造成的总偏差是多少，记为，即当前序列中每个数的真实值为其在序列中的值加。这样新加入一个大小为的数时，相当于在序列中加入一个大小为的数。
对于每个命令，在中插入一个数。
对于每个命令，。
对于每个命令，，再将序列中所有小于的数删除。具体地，将保护结点转到根，将的后继结点转到根的右儿子，将根的右儿子的左儿子整体删除。
对于每个命令，询问第小的值。

Code

#include <bits/stdc++.h>
#define INF 0x7f7f7f7f
#define flag cur->fa->fa!=tar&&cur->fa->fa->s[sn]==cur->fa
using namespace std;
template <class T> inline void read(T &x) {
    x = 0; int c = getchar(), f = 1;
    for (; !isdigit(c); c = getchar()) if (c == 45) f = -1;
    for (; isdigit(c); c = getchar()) (x *= 10) += f*(c-'0');
}
int n, mi, det, tot;
struct SplayNode {
    SplayNode *s[2], *fa; int val, w, sz;
    void update() {sz = w+(s[0]?s[0]->sz:0)+(s[1]?s[1]->sz:0);}
} ;
struct SplayTree {
    SplayNode* rt;
    SplayNode* newnode(int _val) {
        SplayNode* v = new SplayNode;
        v->s[0] = v->s[1] = v->fa = NULL;
        v->val = _val, v->w = v->sz = 1;
        return v;
    }
    SplayTree() {
        rt = newnode(-INF), rt->s[1] = newnode(INF);
        rt->s[1]->fa = rt, rt->update();
    }
    void rotate(SplayNode* v, bool sn) {
        SplayNode* f = v->fa;
        f->s[sn^1] = v->s[sn], v->fa = f->fa;
        if (f->s[sn^1]) f->s[sn^1]->fa = f;
        if (v->fa) v->fa->s[f == f->fa->s[1]] = v;
        v->s[sn] = f, f->fa = v, f->update(), v->update();
    }
    void splay(SplayNode* cur, SplayNode* tar) {
        while (cur != tar && cur->fa != tar) {
            bool sn = cur->fa->s[1] == cur;
            if (flag) rotate(cur->fa, sn^1);
            rotate(cur, sn^1);
        }
        if (cur->fa == tar) rotate(cur, cur->fa->s[0] == cur);
        if (tar == rt) rt = cur;
    }
    SplayNode* predecessor(int _val) {
        SplayNode *cur = rt, *cpy = rt;
        for (; cur; cur = cur->s[_val > cur->val])
            if (cur->val < _val) cpy = cur;
        return cpy;
    }
    SplayNode* successor(int _val) {
        SplayNode *cur = rt, *cpy = rt;
        for (; cur; cur = cur->s[_val >= cur->val])
            if (cur->val > _val) cpy = cur;
        return cpy;
    }
    SplayNode* get_kth(SplayNode* v, int k) {
        int lsz = v->s[0] ? v->s[0]->sz : 0;
        if (k <= lsz) return get_kth(v->s[0], k);
        if (k > lsz+v->w) return get_kth(v->s[1], k-lsz-v->w);
        return v;
    }
    void insert(int _val) {
        if (_val+det < mi) return;
        SplayNode* pre = predecessor(_val);
        SplayNode* suc = successor(_val);
        splay(pre, rt), splay(suc, rt->s[1]);
        if (suc->s[0]) suc->s[0]->w++, suc->s[0]->sz++;
        else suc->s[0] = newnode(_val), suc->s[0]->fa = suc;
        suc->update(), rt->update();
    }
    void remove(int _val) {
        SplayNode* pre = get_kth(rt, 1);
        SplayNode* suc = successor(_val-1);
        splay(pre, rt), splay(suc, rt->s[1]);
        if (suc->s[0]) tot += suc->s[0]->sz;
        suc->s[0] = NULL, suc->update(), rt->update();
    }
} BBST;
int main() {
    read(n), read(mi);
    for (int x; n; n--) {
        char opt[1]; scanf("%s", opt), read(x);
        if (opt[0] == 'I') BBST.insert(x-det);
        if (opt[0] == 'A') det += x;
        if (opt[0] == 'S') BBST.remove(mi-(det-=x));
        if (opt[0] == 'F') {
            x = BBST.rt->sz-x-1;
            if (x < 1 || x > BBST.rt->sz-2) puts("-1");
            else printf("%d\n", BBST.get_kth(BBST.rt, x+1)->val+det);
        }
    }
    return printf("%d\n", tot), 0;
}