HNOI2002 营业额统计 Splay树 单点更新模板

营业额统计

Description

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

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

题解：用splay可以快速地查找到与value值最接近的两个值，然后做一个比较即可。本题用SBT以及set均可解，套一下splay模板练手

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <cmath>
#include <cstdlib>
#include <string>
#include <map>
#include <set>
#include <algorithm>
#include <functional>
#define rep(i,a,b) for (int i=a;i<((b)+1);i++)
#define Rep(i,a,b) for (int i=a;i>=b;i--)
#define foreach(e,x) for (__typeof(x.begin()) e=x.begin();e!=x.end();e++)
#define mid ((l+r)>>1)
#define lson (k<<1)
#define rson (k<<1|1)
#define MEM(a,x) memset(a,x,sizeof a)
#define L ch[r][0]
#define R ch[r][1]
using namespace std;
const int N=1000050;
const long long Mod=1000000007;
const int inf=0x3f3f3f3f;
typedef pair<int, int> pii;
typedef long long ll;
int pre[N],ch[N][2],key[N],root,tot;
void init() {
    root=tot=0;
    ch[root][0]=ch[root][1]=key[root]=pre[root]=0;
}
void newnode(int & r,int fa,int k) {
    r=++tot;
    pre[r]=fa;
    L=R=0;
    key[r]=k;
}
void Rotate(int x,int kind) {
    int y=pre[x];
    ch[y][!kind]=ch[x][kind];
    pre[ch[x][kind]]=y;
    if (pre[y]) ch[pre[y]][ch[pre[y]][1]==y]=x;
    pre[x]=pre[y];
    ch[x][kind]=y;
    pre[y]=x;
}
void splay(int r,int g) {
    while (pre[r]!=g){
        if (pre[pre[r]]==g) Rotate(r, ch[pre[r]][0]==r);
        else {
            int y=pre[r],k=ch[pre[y]][0]==y;
            if (ch[y][k]==r) {
                Rotate(r,!k);
                Rotate(r,k);
            }else {
                Rotate(y,k);
                Rotate(r,k);
            }
        }
    }
    if (g==0)   root=r;
}
void Insert(int k) {
    int r=root;
    if (r==0) {
        newnode(root, 0, k);
        return ;
    }
    while (ch[r][key[r]<k]) r=ch[r][key[r]<k];
    newnode(ch[r][key[r]<k], r, k);
    splay(ch[r][key[r]<k],0);
}
int getmin(int r) {
    while (L)   r=L;
    return r;
}
int getmax(int r) {
    while (R)   r=R;
    return r;
}
int n,x,ans;
int main(int argc, const char * argv[]) {
    scanf("%d",&n);
    rep(i,1,n) {
        scanf("%d",&x);
        Insert(x);
        if (i==1) {
            ans+=x;
            continue;
        }
        int tmp=inf;
        if (ch[root][0])    tmp=min(tmp,key[root]-key[getmax(ch[root][0])]);
        if (ch[root][1])    tmp=min(tmp,key[getmin(ch[root][1])]-key[root]);
        ans+=tmp;
    }
    printf("%d\n",ans);
    return 0;
}