码迷,mamicode.com
首页 > 其他好文 > 详细

Bzoj1503--Noi2004郁闷的出纳员

时间:2016-09-28 09:56:47      阅读:167      评论:0      收藏:0      [点我收藏+]

标签:

treap乱搞一下

对于工资的加减不直接修改的点的值或打标记,而是维护一个外围的delta,这样维护也很方便

剩下删除和插入都是基本操作了

另: linux下rand()的返回值是1-2^31-1,虽然感觉没什么关系,交上去就会被卡T掉。。。取个模就过了

代码:

#include<bits/stdc++.h>
#define MAXN 100205
#define MAXM 3005
#define INF 1000000000
#define MOD 998244353
#define LL long long
using namespace std;

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

int n,m,delta;

struct node{
    int son[2],s,v,r,par,sz;
}x[MAXN];
struct Treap{
    int L,R,root,away,cnt;
    void init() {
        L=0;R=1;root=MAXN-5;away=0;cnt=0;
        x[MAXN-5].r=INF;x[MAXN-5].v=INF;
    }
    
    inline void rotate(int num,int p) {
        int pa=x[num].par,t=x[num].sz;
        x[x[pa].par].son[x[x[pa].par].son[L]==pa?L:R]=num;
        x[num].sz=x[pa].sz;x[pa].sz-=t-x[x[num].son[p]].sz;
        x[pa].son[p^1]=x[num].son[p];if(x[num].son[p]) {x[x[num].son[p]].par=pa;}
        x[num].son[p]=pa;x[num].par=x[pa].par;x[pa].par=num;
    }
    
    void Insert(int v) {
        x[++cnt].v=v;x[cnt].r=rand()%1000007;x[cnt].s=1;x[cnt].sz=1;
        int now=root,pre;
        while(true) {
            if(v==x[now].v) {x[now].s++;x[now].sz++;cnt--;return;}
            pre=v<x[now].v?R:L;x[now].sz++;
            if(!x[now].son[pre]) {x[now].son[pre]=cnt;x[cnt].par=now;break;}
            now=x[now].son[pre];
        }
        now=cnt;
        while(true) {
            if(x[x[now].par].son[L]==now) pre=R;else pre=L;
            if(x[now].r>x[x[now].par].r) rotate(now,pre);
            else break;
        }
    }
    void Del(int num) {
        while(x[num].son[L]) rotate(x[num].son[L],R);
        int now=x[num].par,pre=x[x[num].par].son[L]==num?L:R;
        while(now) {x[now].sz-=x[num].sz;now=x[now].par;}
        away+=x[num].sz;x[x[num].par].son[pre]=0;
    }
    void Check(int v) {
        int now=root,ed;
        while(true) {
            while(x[now].son[R]) now=x[now].son[R];
            if(x[now].v<v) Del(now);
            else break;
            now=root;
        }
    }
    int Qurey(int k) {
        int now=root;
        if(k>x[root].sz) return -1;
        while(true) {
            if(x[x[now].son[L]].sz>=k-x[now].s&&x[x[now].son[L]].sz<k) return x[now].v;
            if(x[x[now].son[L]].sz<k) {k-=x[now].s+x[x[now].son[L]].sz;now=x[now].son[R];}
            else now=x[now].son[L];
            
        }
    }
    
    int Finish() {return away;}
};

int main() {
    srand(2333333);
    n=read();m=read();
    Treap p;char q[10];int k,ret;
    p.init();
    for(int i=1;i<=n;i++) {
        scanf("%s",q);k=read();
        if(q[0]==I) if(k>=m) p.Insert(k+delta);
        if(q[0]==A) delta-=k;
        if(q[0]==S) {delta+=k;p.Check(m+delta);}
        if(q[0]==F) {
            ret=p.Qurey(k);
            printf("%d\n",ret-(ret==-1?0:delta));
        }
    }
    printf("%d\n",p.Finish());
    return 0;
}

 

Bzoj1503--Noi2004郁闷的出纳员

标签:

原文地址:http://www.cnblogs.com/ihopenot/p/5915078.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!