bzoj 1588 [HNOI2002]营业额统计

Description

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

Input

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

Output

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

Sample Input

6
5
1
2
5
4
6   

Sample Output

12

HINT

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

Key To Problem

平衡树求前继和后驱。
在平衡树的分类中查到的题，就写的平衡树，但貌似线段树也能做的样子。。或许我在口胡。

Code

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100010
#define inf 0x3f3f3f3f
using namespace std;
struct node
{
    int l,r,v,rnd,w,size;
};
node a[N];
int n,size,root,ans1,ans2,sum;

void updata(int k)
{
    a[k].size=a[a[k].l].size+a[a[k].r].size+a[k].w;
}

void lturn(int &k)
{
    int t=a[k].r;
    a[k].r=a[t].l,a[t].l=k;
    a[t].size=a[k].size,updata(k),k=t;
}

void rturn(int &k)
{
    int t=a[k].l;
    a[k].l=a[t].r,a[t].r=k;
    a[t].size=a[k].size,updata(k),k=t;
}

void insert(int &k,int x)
{
    if(k==0)
    {
        k=++size;
        a[k].size=a[k].w=1,a[k].v=x,a[k].rnd=rand();
        return ;
    }
    a[k].size++;
    if(x==a[k].v)a[k].w++;
    else if(x>a[k].v)
    {
        insert(a[k].r,x);
        if(a[a[k].r].rnd<a[k].rnd)lturn(k);
    }
    else
    {
        insert(a[k].l,x);
        if(a[a[k].l].rnd<a[k].rnd)rturn(k);
    }
}

void query_before(int &k,int x)
{
    if(k==0)return ;
    if(x>=a[k].v)
        ans1=a[k].v,query_before(a[k].r,x);
    else
        query_before(a[k].l,x);
}

void query_after(int &k,int x)
{
    if(k==0)return ;
    if(x<=a[k].v)
        ans2=a[k].v,query_after(a[k].l,x);
    else
        query_after(a[k].r,x);
}

int main()
{
    cin>>n>>sum;
    insert(root,sum);
    for(int i=2;i<=n;i++)
    {
        int x;
        if(scanf("%d",&x)==EOF)x=0;
        ans1=-inf,ans2=inf;
        query_before(root,x);
        query_after(root,x);
        sum+=min(x-ans1,ans2-x);
        insert(root,x);
    }
    cout<<sum<<endl;
    return 0;
}