BZOJ 1503 郁闷的出纳员 无旋treap

BZOJ 1503

想到这题解法不难

主要是自己debug了半天 竟然因为期间update放循环里面 de了两个多小时 无语 想哭

思路就是用一个tmp代替加减影响 这样我们插入的时候就是插入num-tmp 然后出来的时候 后来更新的结果是tmp' 

tmp' = tmp(后)-tmp(当时) 那是不是你加上这个tmp 就是答案了？

就用无旋转treap解决

/*
bzoj 1503
*/
#include <cstdio>
#include <cmath>
#include <queue>
#include <stack>
#include <algorithm>
#include <cstring>
#include <sstream>
#include <set>
#include <map>
#include <iostream>
#define dbg(x) cout<<#x<<" = "<< (x)<< endl
using namespace std;
const int MAX_N = 500010;


int siz[MAX_N],ch[MAX_N][2],rnd[MAX_N],val[MAX_N];
int T,ans,minn,tmp,cnt,m,x,y,z,rt,p,a;

int read()
{
    int x=  0,f = 1;
    char c= getchar();
    while(c>'9'||c<'0')
    {
        if(c=='-')
            f = -1;
        c = getchar();
    }
    while(c>='0'&&c<='9')
    {
        x = x*10+ c-'0';
        c = getchar();
    }
    return x*f;
}

void update(int x)
{
    siz[x] = 1 + siz[ch[x][0]] + siz[ch[x][1]];
}

int Insert(int a)
{
    siz[++cnt] = 1;
    val[cnt] = a;
    rnd[cnt] = rand();
    return cnt;
}
//区间Merge     if(rnd[A]<rnd[B]) {if(fl[A]) down(A);
//
int Merge(int A,int B){
        if(!A||!B) return A+B;
        if(rnd[A]<rnd[B]) {ch[A][1] = Merge(ch[A][1],B);update(A);return A;}//如果rnd[l]<rnd[r]，我们就保留第一棵树的左子树，另一棵树作为它的右子树；
        else {ch[B][0] = Merge(A,ch[B][0]);update(B);return B;}    //如果rnd[l]>=rnd[r]，那我们可以保留第二棵树的右子树，另一颗树作为它的左子树。
}

void split(int now,int k,int &x,int &y){
    if(!now) x = y = 0;
    else {
        if(k<=siz[ch[now][0]]){
            y = now;
            split(ch[now][0],k,x,ch[now][0]);
        }
        else {
            x = now;
            split(ch[now][1],k-siz[ch[now][0]]-1,ch[now][1],y);
        }
    }
    update(now);
}

int Getrank(int now,int a) //直接按照a-1的权值把树分开，那么x树中最大的应该小于等于a-1，那么a的排名就是size[x]+1
{
    if(!now) return 0;
    if(val[now]>=a) return Getrank(ch[now][0],a);
    return siz[ch[now][0]]+1+Getrank(ch[now][1],a);
}
void Ins(int v){
    int k = Getrank(rt,v);
    split(rt,k,x,y);
    rt = Merge(Merge(x,Insert(v)),y);
}
int kth(int p,int k){
    int rk=ch[p][0]?siz[ch[p][0]]+1:1;
    if(k==rk)return val[p];
    if(k<rk)return kth(ch[p][0],k);
    return kth(ch[p][1],k-siz[ch[p][0]]-1);
}
void del(){
    int k = Getrank(rt,minn - tmp);
    split(rt,k,x,y);
    rt = y;
    ans+=siz[x];
}
int main(){
    m  =read(),minn = read();
    int ans = 0;
    tmp = 0;
    char str[20];
    int opt;
    while(m--){
        scanf("%s%d",str,&opt);
        switch(str[0]){
            case 'I': if(opt>=minn) Ins(opt - tmp);break;
            case 'F': printf("%d\n",opt>siz[rt]?-1:kth(rt,siz[rt]-opt+1)+tmp);break;
            case 'S': tmp-=opt,del();break;
            case 'A': tmp+=opt;break;
        }
    }
    printf("%d\n",ans);
    return 0;
}