BZOJ 1588 [HNOI2002]营业额统计

Splay的插入，找最大，最小值。。。。。

1588: [HNOI2002]营业额统计

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 6966   Solved: 2303
[ 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

Source

[ Submit][ Status]

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int maxn=1000010;
const int INF=~0U>>2;

#define key_valu (ch[ch[root][1]][0])

int ch[maxn][2],key[maxn],pre[maxn];
int tot1,root;

///debug**************************
void ShowTree(int x)
{
    if(x)
    {
        ShowTree(ch[x][0]);
        cout<<"key: "<<key[x]<<" 结点: "<<x<<" 左儿子: "<<ch[x][0]<<" 右儿子: "<<ch[x][1]<<" 父节点: "<<pre[x]<<endl;
        ShowTree(ch[x][1]);
    }
}
///*****************************

void NewNode(int& r,int father,int k)
{
    r=++tot1;
    key[r]=k;
    ch[r][0]=ch[r][1]=0;pre[r]=father;
}

void init()
{
    tot1=root=0;
    memset(ch,0,sizeof(ch));
    memset(pre,0,sizeof(pre));
    memset(key,0,sizeof(key));
}

void rotate(int x,int d)
{
    int y=pre[x];

    ch[y][!d]=ch[x][d];
    pre[ch[x][d]]=y;

    pre[x]=pre[y];
    if(pre[x]) ch[pre[x]][ch[pre[y]][1]==y]=x;

    pre[y]=x;
    ch[x][d]=y;
}

void Splay(int r,int goal)
{
    while(pre[r]!=goal)
    {
        if(pre[pre[r]]==goal)
        {
            rotate(r,ch[pre[r]][0]==r);
        }
        else
        {
            int y=pre[r];
            int kind=(ch[pre[y]][0]==y);

            if(ch[y][kind]==r) rotate(r,!kind);
            else rotate(y,kind);

            rotate(r,kind);
        }
    }
    if(goal==0) root=r;
}

void Insert(int x)
{
    int r=root;
    if(r==0)
    {
        NewNode(root,0,x);
        return ;
    }
    while(ch[r][key[r]<x])
    {
        r=ch[r][key[r]<x];
    }
    NewNode(ch[r][key[r]<x],r,x);
    Splay(ch[r][key[r]<x],0);
}

int Get_Min(int r)
{
    while(ch[r][0]) r=ch[r][0];
    return r;
}
int Get_Max(int r)
{
    while(ch[r][1]) r=ch[r][1];
    return r;
}

int main()
{
    int n,a;
while(scanf("%d",&n)!=EOF)
{
    init();
    int ans=0;
    for(int i=0;i<n;i++)
    {
        if(scanf("%d",&a)==EOF) a=0;
        Insert(a);
        if(!i) ans+=a;
        else
        {
            int temp=INF;
            if(ch[root][0])
                temp=min(temp,a-key[Get_Max(ch[root][0])]);
            if(ch[root][1])
                temp=min(temp,key[Get_Min(ch[root][1])]-a);
            ans+=temp;
        }
    }
    printf("%d\n",ans);
}
    return 0;
}