[HNOI 2002]营业额统计

题意

给定n次操作，每次加入一个新数字，并向答案累加其前驱/后继和其差的较小值。

思路

平衡树裸题，而且我们只需要添加，不需要删除，支持查找前驱，后继。
所以说想法很暴，写起来难度也不大。

代码

#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstring>#include <iostream>using namespace std;const int INF=0x3f3f3f3f;struct Node{    Node *ch[2];    int r,v;    int cmp(int x) const {        if(x==v) return -1;        return x>v;    }};void Rotate(Node* &o, int d) {    Node* k=o->ch[d^1];    o->ch[d^1]=k->ch[d];    k->ch[d]=o;    o=k;}bool Contain(Node* o, int x) {    while(o!=NULL) {        int d=o->cmp(x);        if(d==-1) return true;        else o=o->ch[d];    }    return false;}void Insert(Node* &o,int x) {    if(Contain(o,x)) return;    if(o==NULL) {        o=new Node();        o->ch[0]=o->ch[1]=NULL;        o->v=x;        o->r=rand();    }    else{        int d=o->cmp(x);        Insert(o->ch[d],x);        if(o->ch[d]->r>o->r) Rotate(o, d^1);    }}int findmax(Node* o, int x) {    if (o==NULL) return INF;    if (x>=o->v) return findmax(o->ch[1],x);    else return min(o->v,findmax(o->ch[0],x));}int findmin(Node* o, int x) {    if (o==NULL) return -INF;    if (x<=o->v) return findmin(o->ch[0],x);    else return max(o->v,findmin(o->ch[1],x));}Node *tree;int main() {    int n;    scanf("%d", &n);    int ans=0;    for(int i=1; i<=n; i++) {        int x; scanf("%d", &x);        if(Contain(tree, x)) continue;        Insert(tree, x);        int mx=findmax(tree, x);        int mn=findmin(tree, x);        if(mx+mn==0) ans+=x;        else ans+=min(abs(x-mx),abs(x-mn));    }    printf("%d\n", ans);    return 0;}