bzoj1503 郁闷的出纳员 splay版

自己yy的写法 可能有点奇怪吧 详情看代码 还是蛮短的
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int M=100055,inf=0x3f3f3f3f;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+(c-'0'); c=getchar();}
    return ans*f;
}
int t,root,sum,n,mn,delta,now,leave;
int c[M][2],v[M],size[M],fa[M];
void up(int k){size[k]=size[c[k][0]]+size[c[k][1]]+1;}
void rotate(int x,int& k){
    int y=fa[x],z=fa[y],l=0,r=1;
    if(c[y][1]==x) l=1,r=0;
    if(y==k) k=x;
    else{if(c[z][0]==y) c[z][0]=x; else c[z][1]=x;}
    fa[y]=x; fa[x]=z; fa[c[x][r]]=y;
    c[y][l]=c[x][r]; c[x][r]=y;
    up(y); up(x);
}
void splay(int x,int& k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(c[z][0]==y^c[y][0]==x) rotate(y,k);
            else rotate(x,k);
        }
        rotate(x,k);
    }
}
void insert(int &k,int w,int last){
    if(!k){k=++sum; now=k; fa[k]=last; size[k]=1; v[k]=w; return ;}
    if(w<v[k]) insert(c[k][0],w,k);
    else insert(c[k][1],w,k);
    up(k);
}
void push_before(int k,int w){
    if(!k) return ;
    if(v[k]<w) {t=k; push_before(c[k][1],w);}
    else push_before(c[k][0],w);
}
int find(int x,int rank){
    if(!x) return 0;
    int l=c[x][0],r=c[x][1];
    if(size[l]+1==rank) return x;
    else if(size[l]>=rank) return find(l,rank);
    else return find(r,rank-size[l]-1);
}
int main()
{
    n=read(); mn=read();
    char ch[15]; int k,s;
    while(n--){
        scanf("%s",ch); k=read();
        if(ch[0]=='I'&&k>=mn) insert(root,k-delta,0),splay(now,root);
        if(ch[0]=='A') delta+=k;
        if(ch[0]=='S') {
            delta-=k,t=0,push_before(root,mn-delta); if(!t) continue;
            splay(t,root); int l=c[t][0]; leave+=size[l]+1; root=c[t][1]; fa[root]=0;
        }
        if(ch[0]=='F'){
            if(k>size[root]) printf("-1\n");
            else s=find(root,size[root]-k+1),splay(s,root),printf("%d\n",v[root]+delta);
        }
        //printf("%d ",root); printf("%d [%d %d %d]\n",v[root],size[root],c[root][0],c[root][1]);
    }
    printf("%d\n",leave);
    return 0;
}
View Code
posted @ 2017-05-31 07:40  友人Aqwq  阅读(154)  评论(0编辑  收藏  举报