[BZOJ1588][HNOI2002]营业额统计 无旋Treap

[HNOI2002]营业额统计

时间限制: 5 Sec  内存限制: 162 MB

题目描述

营业额统计 Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。 Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况： 该天的最小波动值 当最小波动值越大时，就说明营业情况越不稳定。 而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。 第一天的最小波动值为第一天的营业额。 ? 输入输出要求

输入

第一行为正整数 ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数(有可能有负数) ，表示第i天公司的营业额。

输出

输出文件仅有一个正整数，即Sigma(每天最小的波动值) 。结果小于2^31 。

样例输入

6
5
1
2
5
4
6	

样例输出

12

提示

结果说明：5+|1-5|+|2-1|+|5-5|+|4-5|+|6-5|=5+4+1+0+1+1=12

该题数据bug已修复.----2016.5.15

 

题解：

这题的思路和之前一道“宠物收养所”很像，我们只需要查找每个数的前驱和后继，然后去做差绝对值较小的即可.

（这道题其实可以STL set水过去，为什么我要打无旋Treap233）

代码见下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <ctime>
#include <cstdlib>
using namespace std;
int n;
struct Treap
{
    Treap *ch[2];
    int val,size,key;
    Treap(){val=size=0;key=rand();ch[0]=ch[1]=NULL;}
    inline void update(){size=1+ch[0]->size+ch[1]->size;}
}*null=new Treap(),*root=null;
typedef pair<Treap*,Treap*> D;
inline Treap* newTreap(int v)
{
    Treap *o=new Treap();
    o->ch[0]=o->ch[1]=null;
    o->val=v;o->size=1;
    return o;
}
Treap *merge(Treap *a,Treap *b)
{
    if(a==null)return b;
    if(b==null)return a;
    if(a->key < b->key)
        {a->ch[1]=merge(a->ch[1],b);a->update();return a;}
    else
        {b->ch[0]=merge(a,b->ch[0]);b->update();return b;}
}
D split(Treap *o,int k)
{
    if(o==null) return D(null,null);
    D y;
    if(o->ch[0]->size >=k)
        {y=split(o->ch[0],k);o->ch[0]=y.second;o->update();y.second=o;}
    else
        {y=split(o->ch[1],k-o->ch[0]->size-1);o->ch[1]=y.first;o->update();y.first=o;}
    return y;
}
int getrank(Treap *o,int v)
{
    if(o==null)return 0;
    return o->val >=v?getrank(o->ch[0],v):getrank(o->ch[1],v)+o->ch[0]->size+1;
}
inline int getval(int rank)
{
    D x=split(root,rank-1);
    D y=split(x.second,1);
    int ret=y.first->val;
    root=merge(merge(x.first,y.first),y.second);
    return ret;
}
inline int query(int val)
{
    int pre=getval(getrank(root,val));
    int back=getval(getrank(root,val)+1);
    return min(abs(back-val),abs(pre-val));
}
inline void insert(int val)
{
    int k=getrank(root,val);
    D x=split(root,k);
    Treap *o=newTreap(val);
    root=merge(merge(x.first,o),x.second);
}
int main()
{
    scanf("%d",&n);int x,ans=0;
    while(n--)
    {
        scanf("%d",&x);
        ans+=query(x);
        insert(x);
    }
    printf("%d",ans);
}