【Treap】BZOJ1588-[HNOI2002]营业额统计

【题目大意】
[借用别人的概括]给出一个n个数的数列a,对于第i个元素ai定义fi=min(abs(ai-aj))，(1<=j<i)，其中f1=a1。输出sum(fi) (1<=i<=n)
【思路】
平衡树，易知绝对值最小的一定是前缀和后继。要注意的是由于ai和aj可能相等，即min可能等于零，要加一个判断：如果当前平衡树中的key等于要找的值，并且当前结点的cnt大于一个，则说明之前有一个值和当前相等。另外有些情况下当前要找的结点可能没有前驱或者后继，需要判断一下。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 const int INF=0x7fffffff;
 struct treap
 {
     int key;
     int priority;
     treap* lson;
     treap* rson;
     int cnt;
     treap()
     {
         key=;
         cnt=;
         /*这里初始值为1*/
         priority=rand();
         lson=rson=NULL;
     }
 };
 treap* root=NULL; 

 void RightRotate(treap* &rt)
 {
     treap* tmp=rt->lson;
     rt->lson=tmp->rson;
     tmp->rson=rt;
     rt=tmp;
 }

 void LeftRotate(treap* &rt)
 {
     treap *tmp=rt->rson;
     rt->rson=tmp->lson;
     tmp->lson=rt;
     rt=tmp;
 }

 void insert(treap* &rt,int x)
 {
     if (rt==NULL)
     {
         rt=new treap;
         rt->key=x;
     }
     else
     {
         if (rt->key==x)
         {
             rt->cnt++;
         }
         else if (rt->key>x)
         {
             insert(rt->lson,x);
             if (rt->lson->priority>rt->priority) RightRotate(rt);
         }
         else
         {
             insert(rt->rson,x);
             if (rt->rson->priority>rt->priority) LeftRotate(rt);
         }
     }
 }

 int pre(treap* &rt,int x)
 {
     int ans=-INF;
     treap* tmp=rt;
     while (tmp)
     {
         if (tmp->key==x && tmp->cnt>) return x; //【Attention】这句话非常重要，千万不要漏掉了！
         if (tmp->key<x)
         {
             ans=max(tmp->key,ans);
             tmp=tmp->rson;
         }
         else tmp=tmp->lson;
     }
     return ans;
 }

 int suc(treap* &rt,int x)
 {
     int ans=INF;
     treap *tmp=rt;
     while (tmp)
     {
         if (tmp->key>x)
         {
             if (tmp->key==x && tmp->cnt>) return x;
             ans=min(tmp->key,ans);//这里手滑写成了max(x,ans)orz
             tmp=tmp->lson;
         }
         else tmp=tmp->rson;
     }
     return ans;
 }

 int result()
 {
     int n,ans;
     if(scanf("%d",&n) == EOF) n = ;
     for (int i=;i<n;i++)
     {
         int turnover;
         if(scanf("%d",&turnover) == EOF) turnover = ;
         insert(root,turnover);
         if (i==)
         {
             ans=turnover;
         }
         else
         {
             int premax,sucmin,ad=INF;
             premax=pre(root,turnover);
             sucmin=suc(root,turnover);
             if (premax!=-INF && premax!=INF) ad=min(ad,turnover-premax);
             if (sucmin!=INF && sucmin!=-INF) ad=min(ad,sucmin-turnover);
             if (ad!=INF && ad!=-INF) ans=ans+ad;
         }
     }
     return ans;
 } 

 int main()
 {
     printf("%d",result());
     return ;
 }