营业额统计[Treap]

题目描述

Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。

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

当最小波动值越大时，就说明营业情况越不稳定。

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

第一天的最小波动值为第一天的营业额。

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

输入格式：

输入由文件’turnover.in’读入。

第一行为正整数n（n<=32767） ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数ai(|ai|<=1000000) ，表示第i天公司的营业额，可能存在负数。

输出格式：

 

 

输入样例#1： 

6
5
1
2
5
4
6

输出样例#1： 

12

说明

结果说明：5+|1-5|+|2-1|+|5-5|+|4-5|+|6-5|=5+4+1+0+1+1=12

---

Treap 边插入边找前驱后继

---

#include<bits/stdc++.h>
#define N 40005
using namespace std;
struct Node{
	int ch[2],key,p,size,num;
}t[N];
int n,a,root,tot,ans1,ans2,ans;
int read(){
    int cnt=0,f=1;char ch=0;
    while(!isdigit(ch)){ch=getchar();if(ch=='-')f=-1;}
    while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
    return cnt*f;
}
int randon(){
	static int seed=233;
	return seed=int(seed*47281LL%2147483647);
}
void update_size(int o){
	t[o].size=t[t[o].ch[0]].size+t[t[o].ch[1]].size+t[o].num;
}
void rotate(int &o,int type){
	int x=t[o].ch[type];
	t[o].ch[type]=t[x].ch[type^1];
	t[x].ch[type^1]=o;
	t[x].size=t[o].size;
	update_size(o);
	o=x; 
}
void insert(int &o,int val){
	if(o==0){
		o=++tot;
		t[o].key=val,t[o].p=randon(),t[o].num=t[o].size=1;
		return;
	}
	t[o].size++;
	if(val==t[o].key){
		t[o].num++;
		return;
	}
	if(val<t[o].key){
		insert(t[o].ch[0],val);
		if(t[t[o].ch[0]].p<t[o].p)
			rotate(o,0);
	}
	else{
		insert(t[o].ch[1],val);
		if(t[t[o].ch[1]].p<t[o].p)
			rotate(o,1); 
	}
}
void pre(int o,int val){
	if(o==0) return;
	if(t[o].key<=val){
		ans1=o,pre(t[o].ch[1],val);
	}
	else pre(t[o].ch[0],val);
}
void suf(int o,int val){
	if(o==0) return;
	if(t[o].key>=val){
		ans2=o,suf(t[o].ch[0],val);
	}
	else suf(t[o].ch[1],val);
}
void Cout(int o){
    if(o==0) return;
    if(t[o].ch[0]) Cout(t[o].ch[0]);
    cout<<t[o].key<<endl;
    Cout(t[o].ch[1]);
}
int main(){
	n=read();
	a=read(),insert(root,a),ans+=a;
	for(int i=2;i<=n;i++){
		a=read();
		pre(root,a),suf(root,a);
		if(ans1==0) ans+=abs(a-t[ans2].key);
		else if(ans2==0) ans+=abs(a-t[ans1].key);
		else ans+=min(abs(t[ans1].key-a),abs(t[ans2].key-a));
		insert(root,a);
	}
	cout<<ans;
	return 0;
}