【NOI2004】郁闷的出纳员 - Splay

题目描述

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

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

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

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

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

思路

用 splay 维护即可，介绍 splay 的文章可参考 『这里』

#include <cstdio>
const int maxn = 100000 + 10;
int n,m,num,ans,delta,root,father[maxn],ch[maxn][2],key[maxn],cnt[maxn],size[maxn];
inline void clear(int x) { ch[x][0] = ch[x][1] = key[x] = cnt[x] = size[x] = father[x] = 0; }
inline bool getfa(int x) { return ch[father[x]][1] == x; }
inline void Update(int x) {
    if (x) {
        size[x] = cnt[x];
        if (ch[x][0]) size[x] += size[ch[x][0]];
        if (ch[x][1]) size[x] += size[ch[x][1]];
    }
}
inline void rotate(int x) {
    int old = father[x],oldfa = father[father[x]],w = getfa(x);
    ch[old][w] = ch[x][w^1]; father[ch[old][w]] = old;
    ch[x][w^1] = old; father[old] = x;
    father[x] = oldfa;
    if (oldfa) ch[oldfa][ch[oldfa][1] == old] = x;
    Update(old);
    Update(x);
}
inline void splay(int x) {
    for (int fa;fa = father[x];rotate(x))
        if (father[fa]) rotate(getfa(fa) == getfa(x) ? fa : x);
    root = x;
}
inline void Insert(int x) {
    if (!root) {
        num++;
        ch[num][0] = ch[num][1] = father[num] = 0;
        key[num] = x;
        cnt[num] = size[num] = 1;
        root = num;
        return;
    }
    int now = root,fa = 0;
    while (true) {
        if (key[now] == x) {
            cnt[now]++;
            Update(now);
            Update(fa);
            splay(now);
            break;
        }
        fa = now;
        now = ch[now][key[now] < x];
        if (!now) {
            num++;
            ch[num][0] = ch[num][1] = 0;
            father[num] = fa;
            cnt[num] = size[num] = 1;
            ch[fa][key[fa] < x] = num;
            key[num] = x;
            Update(fa);
            splay(num);
            break;
        }
    }
}
inline int getnum(int x) {
    int now = root,ans = 0;
    while (true) {
        if (key[now] > x) now = ch[now][0];
        else {
            if (ch[now][0]) ans += size[ch[now][0]];
            if (key[now] == x) {
                splay(now);
                return ans+1;
            }
            ans += cnt[now];
            now = ch[now][1];
        }
    }
}
inline int find(int x) {
    int now = root;
    while (true) {
        if (ch[now][0] && x <= size[ch[now][0]]) now = ch[now][0];
        else {
            int tmp = cnt[now];
            if (ch[now][0]) tmp += size[ch[now][0]];
            if (x <= tmp) return key[now];
            x -= tmp;
            now = ch[now][1];
        }
    }
}
inline void Delete(int x) {
    getnum(x);
    if (cnt[root] > 1) {
        cnt[root]--;
        Update(root);
        return;
    }
    if (!ch[root][0] && !ch[root][1]) {
        clear(x);
        root = 0;
        return;
    } else if (!ch[root][0] || !ch[root][1]) {
        int oldroot = root;
        root = ch[root][!ch[root][0]];
        father[root] = 0;
        clear(oldroot);
        return;
    }
    int l = ch[root][0],oldroot = root;
    while (ch[l][1]) l = ch[l][1];
    splay(l);
    ch[root][1] = ch[oldroot][1];
    father[ch[oldroot][1]] = root;
    clear(oldroot);
    Update(root);
}

int main() {
    scanf("%d%d\n",&n,&m);
    while(n--) {
        char op;
        int k;
        scanf("%s%d",&op,&k);
        if (op == 'I' && k >= m) Insert(k-delta);
        if (op == 'F') printf("%d\n",k <= size[root] ? find(size[root]-k+1)+delta : -1);
        if (op == 'A') delta += k;
        if (op == 'S') {
            int tmp = size[root];
            delta -= k;
            for (int i = 1;i <= tmp;i++)
                if (find(1)+delta < m) {
                    Delete(find(1));
                    ans++;
                }
        }
    }
    printf("%d",ans);
    return 0;
}