[BZOJ1503]郁闷的出纳员 做题笔记

题目来源：http://www.lydsy.com/JudgeOnline/problem.php?id=1503

1、首先，维护一个全局变量del，表示全局加多少。插入时要减去del，查找时要加上del，仔细领会一下。
2、删除时，找到工资比最低限小的工资最高的工人，把他旋转到根，然后删除他和左子树！
3、这题的查找是第K大，而不是第K小，要注意第K大的写法。
4、坑！这题说如果一个工人的初始工资比最低限小，那他就会立即离开，这个离开的工人就当是被UFO抓走了，不算在离开的工人之列！

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=101000;
int n,del=0,low;
int leave=0;
struct Splay {
    int ch[N][2],f[N],root,cnt;
    int a[N],c[N],s[N];
    int grow (int k,int fa) {
        int x=++cnt;
        a[x]=k,s[x]=c[x]=1;
        ch[x][0]=ch[x][1]=0;
        f[x]=fa;
        return x;
    }
    void pushup (int x) {
        s[x]=s[ch[x][0]]+s[ch[x][1]]+c[x];
    }
    void rotate (int x) {
        int y=f[x],opt;
        if (ch[f[x]][0]==x) opt=0;
        else opt=1;
        ch[y][opt]=ch[x][!opt];
        if (ch[x][!opt]) f[ch[x][!opt]]=y;
        f[x]=f[y];
        if (root==y) root=x;
        else if (ch[f[y]][0]==y) ch[f[y]][0]=x;
            else ch[f[y]][1]=x;
        f[y]=x,ch[x][!opt]=y;
        pushup(y),pushup(x);
    }
    void splay (int x,int to=0) {
        while (f[x]!=to) {
            if (f[f[x]]==to) rotate(x);
            else if ((ch[f[f[x]]][0]==f[x])
                ==(ch[f[x]][0]==x))
                rotate(f[x]),rotate(x);
            else rotate(x),rotate(x);
        }
    }
    void insert (int k) {
        if (root==0) root=grow(k,0);
        else {
            int x=root;
            while (x) {
                if (k==a[x]) { c[x]++;break; }
                else {
                    int &y=(k<a[x]?ch[x][0]:ch[x][1]);
                    if (!y) 
                        { x=y=grow(k,x);break; }
                    else x=y;
                }
            }
            splay(x);
        }
    }
    int findval (int k) {//返回k的rank，注意返回值<=rank(k)
        int x=root,ans=0;
        while (x) {
            if (k==a[x]) return ans+s[ch[x][0]]+1;
            else if (k<a[x]) x=ch[x][0];
            else ans+=s[ch[x][0]]+c[x],x=ch[x][1];
        }
        return ans;
    }
    int findkth (int k) {//返回第K大的工资，注意写法
        int x=root;
        while (x) {
            if (k>=s[ch[x][1]]+1&&k<=s[ch[x][1]]+c[x])
                return a[x];
            else if (k>s[ch[x][1]]+c[x]) k-=s[ch[x][1]]+c[x],x=ch[x][0];
            else x=ch[x][1];
        }
        return -1-del;
    }
    int myfind (int k) {//找到rank为k的结点的位置
        int x=root;
        while (x) {
            if (k>=s[ch[x][0]]+1&&k<=s[ch[x][0]]+c[x])
                return x;
            else if (k<s[ch[x][0]]+1) x=ch[x][0];
            else k-=s[ch[x][0]]+c[x],x=ch[x][1];
        }
        return 0;
    }
    void erase (int k) {
        int x=myfind(k);
        splay(x);
        leave+=s[ch[x][0]]+c[x];
        if (ch[x][1]) {
            root=ch[x][1];
            f[ch[x][1]]=0;
            pushup(root);
        }
        else root=cnt=0;
    }
}splay;

int main () {
    char s[10];int x;
    scanf("%d%d",&n,&low);
    for (int i=1;i<=n;i++) {
        scanf("%s %d",s,&x);
        if (s[0]=='I') {
            if (x>=low)  splay.insert(x-del);
            //else leave++;这里是要坑死人啊！
        }
        if (s[0]=='A') del+=x;
        if (s[0]=='S') {
            del-=x;
            int tmp=splay.findval(low-del-1);//找到比最低工资线小的工资最高的工人的排名
            if (tmp) splay.erase(tmp);
        }
        if (s[0]=='F') printf("%d\n",splay.findkth(x)+del);
    }
    printf("%d\n",leave);
}