BZOJ1503: [NOI2004]郁闷的出纳员(splay)

题意：传送门

题解：可以看到一共有四种操作，第一种插入，第二种统一加，第三种统一减，这个副作用是有人出去，第四种是询问工资第K大，可以想到用splay操作，可以将加的操作在外面用个add统一加下，然后见到插入时，如果小于最小工资，直接跳过，否则插入k-add的这么一个员工，对于减，先将外面的add-k，然后找到最小工资minn-add的后一个数，转到根节点下，那么左子树就可以统一处理掉，询问工资第K大，转为询问工资第peo-k+1小即可。最后注意先插入一个工资为inf的员工，因为不插入的话，那么如果只有一个员工，那么这个员工是删不掉的。

附上代码：

#include<bits/stdc++.h>
using namespace std;
#define INF 1000000000
#define MAX 110000
int root,tot,n,mm,ans,add;
struct Node
{
    int ch[2];
    int val;
    int ff;
    int cnt;
    int size;
}t[MAX];
inline void pushup(int x)
{
    t[x].size=t[t[x].ch[0]].size+t[t[x].ch[1]].size+t[x].cnt;
}
inline void rotate(int x)
{
    int y=t[x].ff;
    int z=t[y].ff;
    int k=(x==t[y].ch[1]);
    t[z].ch[y==t[z].ch[1]]=x;
    t[x].ff=z;
    t[y].ch[k]=t[x].ch[k^1];t[t[x].ch[k^1]].ff=y;
    t[x].ch[k^1]=y;t[y].ff=x;
    pushup(y);pushup(x);
}
inline void splay(int x,int goal)
{
    while(t[x].ff!=goal){
        int y=t[x].ff;
        int z=t[y].ff;
        if(z!=goal)(t[z].ch[0]==y)^(t[y].ch[0]==x)?rotate(x):rotate(y);
        rotate(x);
    }
    if(goal==0)root=x;
}
inline void insert(int x)
{
    int u=root,ff=0;
    while(u&&t[u].val!=x){
        ff=u;
        u=t[u].ch[x>t[u].val];
    }
    if(u)t[u].cnt++;
    else{
        u=++tot;
        if(ff)t[ff].ch[x>t[ff].val]=u;
        t[u].ch[0]=t[u].ch[1]=0;
        t[u].val=x;
        t[u].cnt=1;
        t[u].size=1;
        t[u].ff=ff;
    }
    splay(u,0);
}
inline void find(int x)
{
    int u=root;
    if(!u)return;
    while(t[u].ch[x>t[u].val]&&t[u].val!=x)u=t[u].ch[x>t[u].val];
    splay(u,0);
}
inline int Next(int x,int f)
{
    find(x);
    int u=root;
    if(t[u].val>=x&&f)return u;
    if(t[u].val<=x&&!f)return u;
    u=t[u].ch[f];
    while(t[u].ch[f^1])u=t[u].ch[f^1];
    return u;
}
inline int kth(int x)
{
    int u=root;
    if(t[u].size<x||x<=0)return -INF;
    while(233){
        int y=t[u].ch[0];
        if(t[y].size+t[u].cnt<x){
            x-=t[y].size+t[u].cnt;
            u=t[u].ch[1];
        }else
            if(t[y].size>=x)u=y;
            else return t[u].val;
    }
}
int main()
{
    insert(+INF);
    scanf("%d%d",&n,&mm);
    int peo=0;
    while(n--){
        int x;char ch[3];
        scanf("%s%d",ch,&x);
        if(ch[0]=='I'){
            if(x<mm);
            else{
                insert(x-add);
                peo++;
            }
        }
        if(ch[0]=='A')add+=x;
        if(ch[0]=='S'){
            add-=x;
            int gg=Next(mm-add,1);
            splay(gg,0);
            ans+=t[t[root].ch[0]].size;
            peo-=t[t[root].ch[0]].size;
            t[t[root].ch[0]].size=t[t[root].ch[0]].cnt=0;
            t[root].ch[0]=0;
            pushup(root);
        }
        if(ch[0]=='F'){
            int gg=kth(peo-x+1);
            printf("%d\n",gg==-INF?-1:(gg+add));
        }
    }
    printf("%d\n",ans);
    return 0;
}