NOIP 营业额统计 splay tree 纯模板

2924: 营业额统计 

Time Limit(Common/Java):1000MS/3000MS     Memory Limit:65536KByte
Total Submit: 389            Accepted:122

Description

 

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

该天的最小波动值 ＝ min{ |该天以前某天的营业额 － 该天的营业额 | ｝

当最小波动值越大时，就说明营业情况越不稳定。而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。第一天的最小波动值为第一天的营业额。

Input

 

测试数据多组，每组的第一行为正整数n（1 <= n <= 32767), 表示该公司从成立一直到现在的天数. 接下来的n行每行有一个整数Ai(Ai <= 1000000) ， 表示第i天的营业额。处理到EOF为止。

Output

每组数据占一行，每行输出一个整数，每天最小波动值的和。结果小于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

 

思路:在出现过的数字中，找出与当前数字x差值最小的值的差值。求和。

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cstdlib>
using namespace std;
const int N = 1e5+5;
const int INF = 0x3f3f3f3f;
int pre[N] , key[N],ch[N][2],root,tot1;
int n;

void NewNode(int &r ,int father,int k)
{
    r = ++tot1;/**保存的位置**/
    pre[r] = father;
    key[r] = k;
    ch[r][0] = ch[r][1] = 0;
}

void Rotate(int x,int kind)
{
    int y = pre[x]; // x  的父亲节点
    ch[y][!kind] = ch[x][kind]; /**建立x的kind节点和父亲节点的关系**/
    pre[ch[x][kind]] = y;

    if(pre[y]) /**建立x节点和x父亲父亲节点的关系**/
    {
        int k = ch[ pre[y] ] [ 1] == y;
        ch[pre[y]][k] = x;
    }
    pre[x] = pre[y];

    ch[x][kind] = y; /**建立x和y的关系**/
    pre[y] = x;
}
//Splay调整，将结点r调整到goal下面
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;  /**如果是左节点返回1。否则0。取反的kind**/

            if(ch[y][kind] == r)/**r和父亲节点y的位置，和y和其y父亲节点位置不同。**/
            { /**采用 Zig-Zag或Zag-Zig操作**/
                Rotate(r,!kind);
                Rotate(r,kind);
            }
            else/**Zig-Zig或Zag-Zag操作**/
            {
                Rotate(y,kind);
                Rotate(r,kind);
            }
        }
    }
    if(goal == 0) root = r;
}

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

int get_pre(int x)
{
    int temp = ch[x][0];
    if(temp == 0) return INF;
    while(ch[temp][1])
        temp = ch[temp][1];
    return key[x] - key[temp];
}

int get_next(int x)
{
    int temp = ch[x][1];
    if(temp == 0) return INF;
    while(ch[temp][0])
        temp = ch[temp][0];
    return key[temp] - key[x];
}
int main()
{
    int n , x;
    while(scanf("%d",&n)>0)
    {
        root = tot1 = 0;
        int sum = 0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            if(i==1)
            {
                sum = sum + x;
                NewNode(root,0,x);
                continue;
            }
            if(Insert(x) == 0) continue;
            int a = get_pre(root);
            int b = get_next(root);
            sum=sum+min(a,b);
        }
        printf("%d
",sum);
    }
    return 0;
}