【Treap/非旋转Treap】BZOJ1503 [NOI2004]郁闷的出纳员

来源：互联网发布：mac怎么qq远程控制编辑：程序博客网时间：2024/05/16 14:31

题面在这里

Treap的经典题啊……

很典型的Treap，为了练手我用了FHQ_Treap
对于A和S操作，如果一个一个修改，显然是不行的
对于所有Treap中的元素，我们可以设置一个基准线，
所有人工资的真实值为Treap中的key+基准
那么对于A和S操作，我们只需修改基准即可

附上代码：

#include<cstdio>#include<cstdlib>const int maxn=100005;struct node{    node *s[2];    int k,f,siz;    node () {}    node (int x,node *son) {k=x;f=rand();siz=1;s[0]=s[1]=son;}    void pushup() {siz=s[0]->siz + s[1]->siz+1;}}nil,base[maxn];typedef node* P_node;P_node null,len,Root;void Treap_init(){    nil=node(0,NULL);null=&nil;    null->s[0]=null->s[1]=null; null->siz=0;    len=base;Root=null;}P_node newnode(int x){    *len=node(x,null);return len++;}P_node merge(P_node a,P_node b){    if (a==null) return b;    if (b==null) return a;    if (a->f > b->f) {a->s[1]=merge(a->s[1],b);a->pushup();return a;}     b->s[0]=merge(a,b->s[0]);b->pushup();return b;}void split(P_node x,int k,P_node &l,P_node &r){    if (x==null) {l=r=null;return;}    if (x->k <= k) {l=x;split(l->s[1],k,l->s[1],r);}else     {r=x;split(r->s[0],k,l,r->s[0]);}    x->pushup();}int kth(P_node x,int k){    if (k<= x->s[0]->siz) return kth(x->s[0],k);    if (k > x->s[0]->siz+1) return kth(x->s[1],k - x->s[0]->siz -1);    return x->k;}void print(P_node x){    if (x==null) return;    print(x->s[0]);    printf("%d ",x->k);    print(x->s[1]);}int q,Min,sta,ans=0;inline int red(){    int tot=0,f=1;char ch=getchar();    while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=getchar();}    while ('0'<=ch&&ch<='9') tot=tot*10+ch-48,ch=getchar();    return tot*f;}int main(){    Treap_init();    q=red();Min=red();sta=0;    while (q--){        char c=getchar();while (c<'A'||'Z'<c) c=getchar();        if (c=='I'){            int k=red()-sta;if (k<Min-sta) continue;            P_node l,r; split(Root,k,l,r);            Root=merge(l,merge(newnode(k),r));        }else        if (c=='A') sta+=red();else        if (c=='S'){            int k=red();P_node tem;            sta-=k; split(Root,Min-sta-1,tem,Root);            ans+=tem->siz;        }else{            int k=red();            if (k>Root->siz) printf("-1\n");else            printf("%d\n",kth(Root,Root->siz-k+1)+sta);        }    }    printf("%d\n",ans);    return 0;}

3 0