1588: [HNOI2002]营业额统计 (splay模板)

1588: [HNOI2002]营业额统计

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 5860   Solved: 1892
[ Submit][ Status]

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

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 100001
#define INF 0x3f3f3f3f
#define LL long long
#define clr(x,v); memset(x,v,sizeof(x));
struct SplayTree
{
    int ch[maxn][2],pre[maxn];
    int sz[maxn],val[maxn];
    int size;
    int root;
    ///旋转操作，c=0左旋，c=1右旋
    inline void Rotate(int x,int c)
    {
        int y=pre[x];
        Push_Down(y),Push_Down(x);
        ch[y][!c]=ch[x][c];
        pre[ch[x][c]]=y;
        pre[x]=pre[y];
        if(ch[pre[y]][0]==y) ch[pre[y]][0]=x;
        else ch[pre[y]][1]=x;
        ch[x][c]=y,pre[y]=x;
        Push_Up(y);
    }
    ///将结点x转到结点tag的下面
    inline void Splay(int x,int tag)
    {
        Push_Down(x);
        while(pre[x]!=tag)
        {
            if(pre[pre[x]]==tag)
            {
                if(ch[pre[x]][0]==x) Rotate(x,1);
                else Rotate(x,0);
            }
            else
            {
                int y=pre[x],z=pre[y];
                if(ch[z][0]==y)
                {
                    if(ch[y][0]==x)
                        Rotate(y,1),Rotate(x,1);
                    else Rotate(x,0),Rotate(x,1);
                }
                else
                {
                    if(ch[y][0]==x)
                        Rotate(x,1),Rotate(x,0);
                    else Rotate(y,0),Rotate(x,0);
                }
            }
        }
        Push_Up(x);
        if(!tag) root=x;
    }
    inline void Init()
    {
        ch[0][0]=ch[0][1]=pre[0]=sz[0]=0;
        root=size=0;
        Newnode(0,root,-INF);
        Newnode(root,ch[root][1],INF);
        sz[root]=2;
    }
    inline void Newnode(int y,int &x,int v)
    {
        x=++size;
        ch[x][0]=ch[x][1]=0,pre[x]=y;
        sz[x]=1;
        //cnt[x]=1;
        val[x]=v;
    }
    inline void vist(int x)
    {
        if(x)
        {
            printf("node%2d : left child  %2d right child  %2d   %2d sz=%d\n",x,ch[x][0],ch[x][1],val[x],sz[x]);
            vist(ch[x][0]);
            vist(ch[x][1]);
        }
    }
    ///找到中序遍历第k个结点，并旋转到tag结点下面
    inline void Select(int k,int tag)
    {
        int t=root,tmp;
        Push_Down(t);
        while((tmp=sz[ch[t][0]])!=k-1)
        {
            if(k<tmp+1) t=ch[t][0];
            else k-=(tmp+1),t=ch[t][1];
            Push_Down(t);
        }
        Splay(t,tag);
    }
    inline int Pre(int key)
    {
        int x=root,res=-INF;
        while(x)
        {
            if(val[x]==key) return key;
            if(val[x]<key) res=max(res,val[x]),x=ch[x][1];
            else x=ch[x][0];
        }
        return res;
    }
    inline int Suc(int key)
    {
        int x=root,res=INF;
        while(x)
        {
            if(val[x]==key) return key;
            if(val[x]>key) res=min(res,val[x]),x=ch[x][0];
            else x=ch[x][1];
        }
        return res;
    }
    inline void Insert(int key)
    {
        int x=root;
        while(ch[x][val[x]<key]) x=ch[x][val[x]<key];
        Newnode(x,ch[x][val[x]<key],key);
        Splay(size,0);
    }
    inline void Push_Down(int x)
    {

    }
    inline void Push_Up(int x)
    {

    }
} spt;
int main()
{
    int n;
    scanf("%d",&n);
    spt.Init();
    int a;
    scanf("%d",&a);
    int ans=0;
    ans=a;
    spt.Insert(a);
    while(--n)
    {
        if(scanf("%d",&a)==-1) a=0;
        ans+=min(a-spt.Pre(a),spt.Suc(a)-a);
        spt.Insert(a);
    }
    printf("%d\n",ans);
    return 0;
}