洛谷 1486 郁闷的出纳员【Treap】

【题意概述】

　　要求维护一个序列支持以下操作：

　　1，插入元素x；

　　2，把序列的所有元素加上x；

　　3，把序列的所有元素减去x，同时低于一个给定的下限的元素马上被删除；

　　4，询问序列中第k大的元素。

【题解】

　　Treap

　　用一个delta记录元素的增减情况，即实际上的元素的值应该是qval(root,x)+delta，那么插入元素时插入的应该是Val-delta.

　　每次加减操作的时候改delta和下限Minwage，加操作的时候Minwage-=val（即下限相对于元素值减小了Val），减操作的时候Minwage+=val（即下限相对于元素值增加了Val）

　　每次减操作时分裂出序列里小于Minwage的元素，把它们扔掉即可。

#include<cstdio>
#include<algorithm>
#define ls (a[u].l)
#define rs (a[u].r)
using namespace std;
const int maxn=200010;
int n,k,x,y,z,v,tot,root,Minwage,Minnow,delta;
char c,c2;
struct treap{int l,r,v,rnd,size;}a[maxn];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
void newnode(int val){a[++tot]=(treap){0,0,val,rand()+1,1};}
void update(int u){a[u].size=a[ls].size+a[rs].size+1;}
void split(int u,int k,int &x,int &y){
    if(!k){x=0; y=u; return;}
    if(a[u].size==k){x=u; y=0; return;}
    if(a[ls].size>=k) split(ls,k,x,ls),y=u;
    else split(rs,k-a[ls].size-1,rs,y),x=u;
    update(u);
}
int merge(int x,int y){
    if(!x||!y) return x+y;
    if(a[x].rnd<a[y].rnd){a[x].r=merge(a[x].r,y); update(x); return x;}
    else{a[y].l=merge(x,a[y].l); update(y); return y;}
}
int qrank(int u,int val){
    if(!u) return 0;
    return a[u].v>=val?qrank(ls,val):qrank(rs,val)+a[ls].size+1;
}
int qval(int u,int k){
    if(a[ls].size+1==k) return a[u].v;
    return a[ls].size>=k?qval(ls,k):qval(rs,k-a[ls].size-1);
}
int main(){
    srand(20020705);
    n=read(); Minwage=Minnow=read();
    while(n--){
        c=getchar();
        while(c!='I'&&c!='A'&&c!='S'&&c!='F') c=getchar();
        c2=getchar(); v=read();
        if(c=='I'&&v>=Minwage){
            split(root,qrank(root,v-delta),x,y);
            newnode(v-delta); root=merge(merge(x,tot),y);
        }
        if(c=='A') Minnow-=v,delta+=v;
        if(c=='S'){
            Minnow+=v; delta-=v;
            split(root,qrank(root,Minnow),x,y);
            root=y;
        }
        if(c=='F') printf("%d\n",a[root].size>=v?qval(root,a[root].size-v+1)+delta:-1);
    }
    return printf("%d",tot-a[root].size),0;
}