BZOJ1588[HNOI2002]营业额统计

https://www.lydsy.com/JudgeOnline/problem.php?id=1588

最后一道Treap模板题了；

就是直接求前驱和后继，就可以了，没什么多说的。

#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].sz=t[tot].cnt=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].r||t[p].l){
            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;
    }
    val<t[p].val?Remove(t[p].l,val):Remove(t[p].r,val);
    update(p);
}
ll GetPre(ll val)
{
    int ans=1,p=root;
    while(p!=0){
        if(val==t[p].val){
            ans=p;
            break;
        }
        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;
}
ll GetNext(ll val)
{
    int ans=2,p=root;
    while(p!=0){
        if(val==t[p].val){
            ans=p;
            break;
        }
        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);
    ll ans=0,x;
    build();
    scanf("%lld",&x);ans+=x;
    Insert(root,x);
    for(int i=1;i<n;i++){
        scanf("%lld",&x);
        ///cout<<GetPre(x)<<" "<<GetNext(x)<<endl;
        ans+=min(x-GetPre(x),GetNext(x)-x);
        Insert(root,x);
    }
    printf("%lld",ans);
    return 0;
}