bzoj 1588 营业额统计 （treap 数组实现）

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

最小波动值= min { | 该天以前某一天的营业额-该天的营业额 | }

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

思路
听说单纯的BST会T掉，所以用treap。刚刚学，代码不优。
rand旋转，玄学稳定树的形状，（成链的极限状态概率很小）。
比较前趋后继取小。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#define LL long long
using namespace std;

int idc, ans, n, x, root;
LL sum;

struct Node{
    int ls, rs, size, w, val, rnd;
}ed[1000005];

void update(int k){
    ed[k].size = ed[ed[k].ls].size + ed[ed[k].rs].size + ed[k].w;
}

void lturn(int &k){//左旋 
    int t = ed[k].rs; ed[k].rs = ed[t].ls; ed[t].ls = k;
    ed[t].size = ed[k].size; update(k); k = t;//要修改k，所以用& 
}

void rturn(int &k){//右旋 
    int t = ed[k].ls; ed[k].ls = ed[t].rs; ed[t].rs = k;
    ed[t].size = ed[k].size; update(k); k = t;//
}

void query_pre(int k, int x){//前趋 
    if(k == 0) return;//到叶子节点后return 
    if(ed[k].val == x) {
        ans = x;
        return;
    }
    if(ed[k].val < x){
        ans = ed[k].val;//
        query_pre(ed[k].rs, x);
    }    
    else
        query_pre(ed[k].ls, x);
}

void query_sub(int k, int x){//后继 
    if(k == 0) return;//
    if(ed[k].val == x){
        ans = x;
        return;
    }
    if(ed[k].val > x){
        ans = ed[k].val;
        query_sub(ed[k].ls, x);
    }
    else
        query_sub(ed[k].rs, x);
}

void insert(int &k, int x){
    if(k == 0){//新建节点
        k = ++idc;//
        ed[k].rnd = rand();
        ed[k].val = x;
        ed[k].size = ed[k].w = 1;
        return;
    }
    ++ed[k].size;//新加节点在当前k的子树中
    if(ed[k].val == x){
        ++ed[k].w;
        return;
    }
    if(x > ed[k].val) {
        insert(ed[k].rs, x);
        if(ed[ed[k].rs].rnd < ed[k].rnd) lturn(k);//旋转维护树形
    }
    else{
        insert(ed[k].ls, x);
        if(ed[ed[k].ls].rnd < ed[k].rnd) rturn(k);
    }
}

int main(){
    scanf("%d", &n);
    scanf("%d", &x);
    sum += (LL) x;//第一个
    insert(root, x);   
    for(int i=2; i<=n; i++){
        scanf("%d", &x);
        ans = -1000000000;
        query_pre(root, x);
        int pre = abs(x - ans);
        ans = -1000000000;
        query_sub(root, x);
        int sub = abs(x - ans);
        sum += (LL) min(pre, sub);
        insert(root, x);//先求值再插入，改变了root的值，不为0 
    }
    printf("%lld\n", sum);
    return 0;
}