BZOJ1208 [HNOI2004]宠物收养所

BZOJ1208

中文题意不怎么多说了。

做法：这道题首先可以用set做。这里就不说了，我们选择用Treap来做，用来练手。

直接在Treap中插入，寻找前驱，后继进行比较就可以了。

你可以开两个Treap，但不过仔细一想，某一时刻只能有多余的宠物或者多余的领养者。所以这道题一个Treap是可以的。

但不过刚刚开始由于范围开大了，T了好几发。所以根据题意来开范围就可以了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace io {
    const int SIZE = 1e7 + 10;
    char inbuff[SIZE];
    char *l, *r;
    inline void init() {
        l = inbuff;
        r = inbuff + fread(inbuff, 1, SIZE, stdin);
    }
    inline char gc() {
        if (l == r) init();
        return (l != r) ? *(l++) : EOF;
    }
    void read(int &x) {
        x = 0; char ch = gc();
        while (!isdigit(ch)) ch = gc();
        while (isdigit(ch)) x = x * 10 + ch - '0', ch = gc();
    }
} using io::read;
const ll inf=INT_MAX;
const int N=1e5+10;
const int M=20;
const ll mod=1000000;
struct Treap
{
    int l,r;
    int dat;
    ll val;
    int cnt,sz;
}t[N];
int tot,root,n;
void init(){tot=0;}
int newnode(ll val)
{
    t[++tot].val=val;t[tot].dat=rand();
    t[tot].l=t[tot].r=0;
    t[tot].cnt=t[tot].sz=1;
    return tot;
}
void update(int p){t[p].sz=t[t[p].l].sz+t[t[p].r].sz+t[p].cnt;}
void build(){newnode(-inf);newnode(inf);root=1,t[1].r=2;update(root);}
void zig(int &p){int q=t[p].l;t[p].l=t[q].r,t[q].r=p,p=q;update(t[p].r),update(p);}
void zag(int &p){int q=t[p].r;t[p].r=t[q].l,t[q].l=p,p=q;update(t[p].l),update(p);}
void Insert(int &p,ll val)
{
    if(p==0){
        p=newnode(val);
        return;
    }
    if(val==t[p].val){
        t[p].cnt++,update(p);
        return;
    }
    if(val<t[p].val){
        Insert(t[p].l,val);
        if(t[p].dat<t[t[p].l].dat) zig(p);
    }
    else{
        Insert(t[p].r,val);
        if(t[p].dat<t[t[p].r].dat) zag(p);
    }
    update(p);
}
void Remove(int &p,ll val)
{
    if(p==0) return;
    if(val==t[p].val){
        if(t[p].cnt>1){
            t[p].cnt--,update(p);
            return;
        }
        if(t[p].l||t[p].r){
            if(t[p].r==0||t[t[p].l].dat>t[t[p].r].dat)
                zig(p),Remove(t[p].r,val);
            else zag(p),Remove(t[p].l,val);
            update(p);
        }
        else p=0;
        return;
    }
    if(val<t[p].val) Remove(t[p].l,val);
    else Remove(t[p].r,val);
    update(p);
}
int GetPre(ll val)
{
    int ans=1,p=root;
    while(p!=0){
        if(t[p].val==val){
            return t[p].val;
        }
        if(t[p].val<val&&t[p].val>t[ans].val) ans=p;
        p=val<t[p].val?t[p].l:t[p].r;
    }
    return t[ans].val;
}
int GetNext(ll val)
{
    int ans=2,p=root;
    while(p!=0){
        if(t[p].val==val){
            return t[p].val;
        }
        if(t[p].val>val&&t[p].val<t[ans].val) ans=p;
        p=val<t[p].val?t[p].l:t[p].r;
    }
    return t[ans].val;
}
int main()
{
    scanf("%d",&n);
    int op,cnt1=0,cnt2=0;
    ll ans=0;
    build();
    ll p1,p2,a;
    for(int i=1;i<=n;i++){
        scanf("%d%lld",&op,&a);
        if(op==0){
            if(cnt2!=0){
                cnt2--;
                p1=GetPre(a),p2=GetNext(a);
                if(p2-a<a-p1) ans+=p2-a,Remove(root,p2);
                else ans+=a-p1,Remove(root,p1);
            }
            else{
                cnt1++,Insert(root,a);
            }
        }
        else{
            if(cnt1!=0){
                cnt1--;
                p1=GetPre(a),p2=GetNext(a);
                if(p2-a<a-p1) ans+=p2-a,Remove(root,p2);
                else ans+=a-p1,Remove(root,p1);
            }
            else{
                cnt2++,Insert(root,a);
            }
        }
        ans%=mod;
    }
    printf("%lld\n",ans);
    return 0;
}