[NOI2004] 郁闷的出纳员 - 平衡树

《郁闷的出纳员》此题曾一度令人十分郁闷,原因是全局标记的转化容易出错。
可以将数值视作一维系量,则有 r=r0+r ,其中 r 为真实工资,r0为系工资, r 为相对工资。平衡树中储存的便是 ri ,另建立一个全局标记 r0 ,则工资加减可视作对它的操作。但此时,出入平衡树的参量都需经过关于 r0 的运算。这是一个需要细心与耐心的过程。

    // 平衡树中记录相对工资,实际为绝对工资,a0为系工资
    //   绝对工资=系工资+相对工资
    //  增加全部工资时,实际上是增加了绝对工资,相对工资不变,因此系工资增加
    //  获取工资时,获取的是绝对工资,有绝对工资=系工资+相对工资
    //  放入工资时,放入的是绝对工资,有相对工资=绝对工资-系工资 
    //  low中记录的是最低的相对工资,最低绝对工资=系工资+最低相对工资 

下面给出参考代码。

#include<cstdio>
using namespace std;
int fa[100005],ch[100005][2],k[100005],cnt[100005],size[100005],m,ind=0,root=0,t1,t2,ans=0,n,low,a0=0;
void update(int p){ch[0][0]=ch[0][1]=size[0]=0;size[p]=size[ch[p][0]]+size[ch[p][1]]+cnt[p];}
void rotate(int p){int q=fa[p],y=fa[q],x=ch[q][1]==p;ch[q][x]=ch[p][x^1];fa[ch[q][x]]=q;ch[p][x^1]=q;fa[q]=p;fa[p]=y;if(y)if(ch[y][0]==q)ch[y][0]=p;else if(ch[y][1]==q)ch[y][1]=p;update(q);update(p);}
void splay(int x){for(int y;y=fa[x];rotate(x))if(fa[y])rotate((x==ch[y][0])==(y==ch[fa[y]][0])?y:x);root=x;}
int search(int v){int p=root;for(;p&&k[p]!=v;p=ch[p][k[p]<v]);if(k[p]==v)return p;else return 0;}
void insert(int v){if(!root)root=++ind,k[root]=v,cnt[root]=size[root]=1,fa[root]=ch[root][0]=ch[root][1]=0;else if(search(v))splay(search(v)),cnt[root]++,size[root]++;else{int x=root,y;for(;y=ch[x][k[x]<v];x=y);y=++ind;k[y]=v;cnt[y]=size[y]=1;fa[y]=x;ch[y][0]=ch[y][1]=0;ch[x][k[x]<v]=y;splay(y);}}
int kth(int _k){int p=root,k=_k;for(;p;)if(ch[p][0]){if(k>size[ch[p][0]]&&k<=size[ch[p][0]]+cnt[p])return p;if(k<=size[ch[p][0]])p=ch[p][0];else k-=(size[ch[p][0]]+cnt[p]),p=ch[p][1];}else if(k<=cnt[p])return p;else k-=cnt[p],p=ch[p][1];return 0;}
void refresh(int rf){
    insert(rf);
    splay(search(rf));
    ans+=size[ch[root][0]]+cnt[root]-1;
    root=ch[root][1];
    fa[root]=0;
}
int main(){
    scanf("%d%d",&n,&low);
    for(int i=0;i<n;i++){
        int t;char c[10];
        scanf("%s%d",c,&t);
        switch(c[0]){
            case 'I':if(t>=low)insert(t-a0);break;
            case 'A':a0+=t;refresh(low-a0-1);break;
            case 'S':a0-=t;refresh(low-a0-1);break;
            case 'F':printf(" %d\n",(t>size[root])?-1:k[kth(size[root]-t+1)]+a0);break;
        }
    }
    printf(" %d\n",ans);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值