treap 1286郁闷的出纳员.cpp

#include<cstdio>
#include<cstdlib>
#include<ctime>
struct shu
{
int l,r,sum,zhi,dui;
}a[100005];
int n,xia,root,size,lei,leave;
char ch[2];
void you(int &a1)
{
int t=a[a1].l;
a[a1].l=a[t].r;
a[t].r=a1;
a[t].sum=a[a1].sum;
a[a1].sum=a[a[a1].l].sum+a[a[a1].r].sum+1;
a1=t;
return;
}
void zuo(int &a1)
{
int t=a[a1].r;
a[a1].r=a[t].l;
a[t].l=a1;
a[t].sum=a[a1].sum;
a[a1].sum=a[a[a1].l].sum+a[a[a1].r].sum+1;
a1=t;
return;
}
void jia(int &a1,int a2)
{
if(a1==0)
{
size++;
a1=size;
a[a1].sum=1;
a[a1].zhi=a2;
a[a1].dui=rand();
return;
}
a[a1].sum++;
if(a2<a[a1].zhi)
{
jia(a[a1].l,a2);
if(a[a[a1].l].dui<a[a1].dui)
you(a1);
}
else
{
jia(a[a1].r,a2);
if(a[a[a1].r].dui<a[a1].dui)
zuo(a1);
}
return;
}
int shan(int &a1,int a2)
{
if(a1==0)
return 0;
int t;
if(a[a1].zhi<a2)
{
t=a[a[a1].l].sum+1;
a1=a[a1].r;
return t+shan(a1,a2);
}
t=shan(a[a1].l,a2);
a[a1].sum-=t;
return t;
}
int zhao(int a1,int a2)
{
if(a[a[a1].r].sum+1==a2)
return a[a1].zhi;
if(a[a[a1].r].sum>=a2)
return zhao(a[a1].r,a2);
else
return zhao(a[a1].l,a2-a[a[a1].r].sum-1);
}
int main()
{
root=size=0;
srand(time(0));
scanf("%d%d",&n,&xia);
for(int i=0;i<n;i++)
{
int a1;
scanf("%s%d",ch,&a1);
if(ch[0]=='I')
if(a1>=xia)
jia(root,a1-lei);
if(ch[0]=='A')
lei+=a1;
if(ch[0]=='S')
{
lei-=a1;
leave+=shan(root,xia-lei);
}
if(ch[0]=='F')
{
if(a1>a[root].sum)
printf("-1\n");
else
printf("%d\n",zhao(root,a1)+lei);
}
}
printf("%d\n",leave);
return 0;
}