NOI 2002 营业额统计 (splay or fhq treap)

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<cstdio>
#include<iostream>
#define inf 0x3f3f3f3f
using namespace std;
int n,root,sz,key[],pre[],ch[][];
void newnode(int &r,int f,int k)
{
     r=++sz;
     pre[r]=f;
     key[r]=k;
     ch[r][]=ch[r][]=;
}
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]][]==y]=x;
     pre[x]=pre[y];
     ch[x][kind]=y;
     pre[y]=x;
}
void splay(int r,int goal)
{
     while(pre[r]!=goal){
                         if(pre[pre[r]]==goal)rotate(r,ch[pre[r]][]==r);
                         else{
                              int y=pre[r];
                              int kind=ch[pre[y]][]==y;
                              if(ch[y][kind]==r){
                                                 rotate(r,!kind);
                                                 rotate(r,kind);}
                              else{
                                   rotate(y,kind);
                                   rotate(r,kind);}}}
     if(!goal)root=r;
}
void insert(int k)
{
    int r=root;
    if(!r){
              newnode(root,,k);
              return;}
    while(ch[r][key[r]<k])r=ch[r][key[r]<k];
    newnode(ch[r][k>key[r]],r,k);
    splay(ch[r][k>key[r]],);
}
int get_next(int x)
{
    int tmp=ch[x][];
    if(!tmp)return inf;
    while(ch[tmp][])tmp=ch[tmp][];
    return key[tmp]-key[x];
}
int get_pre(int x)
{
    int tmp=ch[x][];
    if(!tmp)return inf;
    while(ch[tmp][])tmp=ch[tmp][];
    return key[x]-key[tmp];
}
int main()
{
    scanf("%d",&n);
    int num,ans=;
    for(int i=;i<=n;i++){
            scanf("%d",&num);
            insert(num);
            if(i==)ans+=num;
            else{
                 int a=get_next(root),b=get_pre(root);
                 ans+=min(a,b);}
            }
    printf("%d\n",ans);
    return ;
}

更新fhq treap模板哦～～～

#include<cstdio>
#include<cstdlib>
#include<climits>
#include<iostream>
#define INF INT_MAX/3*2
using namespace std;
inline int Rin(){
  int x=,c=getchar(),f=;
  for(;c<||c>;c=getchar())
    if(!(c^))f=-;
  for(;c>&&c<;c=getchar())
    x=(x<<)+(x<<)+c-;
  return x*f;
}
struct Seg{
  struct node{
    node*ch[];
    int v,s,r,size;
    node(int v,node*_):v(v){s=size=;r=rand();ch[]=ch[]=_;}
    void pu(){size=ch[]->size+ch[]->size+s;}
  }*rt,*bf;
  Seg(){
    bf=new node(,);bf->r=INF;
    bf->s=bf->size=;
    bf->ch[]=bf->ch[]=bf;
    rt=bf;
  }
  void rot(node*&o,int d){
    node*k=o->ch[d];o->ch[d]=k->ch[d^];k->ch[d^]=o;
    o->pu();k->pu();o=k;
  }
  void ins(node*&o,int x){
    if(o==bf){o=new node(x,bf);return;}
    if(o->v==x){o->s++;o->size++;return;}
    ins(o->ch[x>o->v],x);
    if(o->ch[x>o->v]->r<o->r)rot(o,x>o->v);
    else o->pu();
  }
  int prev(node*o,int x){
    if(o==bf)return -INF;
    if(o->v==x&&o->s>)return o->v;
    if(x<=o->v)return prev(o->ch[],x);
    return max(o->v,prev(o->ch[],x));
  }
  int nxt(node*o,int x){
    if(o==bf)return INF;
    if(o->v==x&&o->s>)return o->v;
    if(x>=o->v)return nxt(o->ch[],x);
    return min(o->v,nxt(o->ch[],x));
  }
}T;
int n,x,nxt,prev,ans;
int main(){
  srand('z'+'t'+'k'+'s'+'q');
  n=Rin();
  for(int i=;i<=n;i++){
    x=Rin();T.ins(T.rt,x);
    if(!(i^)){ans=x;continue;}
    prev=x-T.prev(T.rt,x),nxt=T.nxt(T.rt,x)-x;
    ans+=min(prev,nxt);
  }
  printf("%d\n",ans);
  return ;
}