【NOI】2004 郁闷的出纳员

【算法】平衡树(treap)

【题解】

treap知识见数据结构。

解法，具体细节见程序。

#include<cstdio>
#include<algorithm>
#include<ctime>
using namespace std;
const int maxn=;
struct cyc{int l,r,rnd,s,num;}t[maxn*];
int n,sz,low,root,delta;
void rturn(int &tt)
{
    int k=t[tt].l;
    t[tt].l=t[k].r;
    t[k].r=tt;
    t[k].s=t[tt].s;
    t[tt].s=t[t[tt].l].s+t[t[tt].r].s+;
    tt=k;
}
void lturn(int &tt)
{
    int k=t[tt].r;
    t[tt].r=t[k].l;
    t[k].l=tt;
    t[k].s=t[tt].s;
    t[tt].s=t[t[tt].l].s+t[t[tt].r].s+;
    tt=k;
}
void insert(int &k,int x)
{
    if(k==)
     {
         k=++sz;
         t[k].rnd=rand();
         t[k].s=;//s表示该子树的节点个数(含本身)
         t[k].num=x;
         return;
     }
    t[k].s++;
    if(x<t[k].num)
     {
         insert(t[k].l,x);
         if(t[t[k].l].rnd<t[k].rnd)rturn(k);
     }
    else
     {
         insert(t[k].r,x);
         if(t[t[k].r].rnd<t[k].rnd)lturn(k);
     }
}
int del(int &k,int x)
{
    int sum=;
    if(k==)return ;
    if(x>t[k].num){sum=t[t[k].l].s+;k=t[k].r;return sum+del(k,x);}//直接将根从k变为k的右子树，由于&的传递效果，相当于直接废了左子树和根节点，把右子树接上去。
     else{sum=del(t[k].l,x);t[k].s-=sum;return sum;}
}
int find(int k,int x)
{
    if(t[t[k].l].s+==x)return(t[k].num+delta);
    if(x<=t[t[k].l].s)return(find(t[k].l,x));
     else return(find(t[k].r,x-t[t[k].l].s-));
}
int main()
{
    scanf("%d%d",&n,&low);
    srand(time());
    int ans=;
    for(int i=;i<=n;i++)
     {
         char c=getchar();int rd;
         while(c<'A'||c>'Z')c=getchar();
         scanf("%d",&rd);
         if(c=='I')if(rd>=low)insert(root,rd-delta);
         if(c=='A')delta+=rd;
         if(c=='S'){delta-=rd;ans+=del(root,low-delta);}
         if(c=='F')
         {
             if(t[root].s<rd)printf("-1\n");
              else printf("%d\n",find(root,t[root].s-rd+));
         }
     }
    printf("%d",ans);
    return ;
}