洛谷P2234 [HNOI2002]营业额统计_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

上次发了 Splay ，这次再发一个 Treap 。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAX 999999999
using namespace std;
int answer=0;
struct node{
	node* son[2];
	int v,w,s,flag;
	node(){
		son[0]=son[1]=NULL;
		w=rand();
		v=0;
		s=flag=1;
	}
};
node* rt;
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
inline int abs(const int x){if(x<0)return -x;return x;}
inline int min(const int x,const int y){if(x<y)return x;return y;}
void maintain(node* &u){
	u->s=u->flag;
	if(u->son[0]!=NULL)u->s+=u->son[0]->s;
	if(u->son[1]!=NULL)u->s+=u->son[1]->s;
}
void turn(node* &u,int f){
	node* t=u->son[f^1];
	u->son[f^1]=t->son[f];
	t->son[f]=u;
	maintain(u);
	maintain(t);
	u=t;
}
bool  insert(node* &u,int x){
	if(u==NULL){
		u=new node;
		u->v=x;
		return false;
	}
	else if(u->v==x){
		u->flag++;
		maintain(u);
		return true;
	}
	int y=x>u->v?1:0;
	bool f=insert(u->son[y],x);
	if(u->son[y]->w>u->w)turn(u,y^1);
	else maintain(u);
	return f;
}
void front(node* u,int k,int &ans){
	if(u==NULL)return;
	if(u->v<k){
		if(u->v>ans)ans=u->v;
		if(u->son[1]!=NULL)front(u->son[1],k,ans);
	}
	else if(u->v==k&&u->flag>1){
		ans=u->v;
		return;
	}
	else if(u->v>=k)
	if(u->son[0]!=NULL)front(u->son[0],k,ans);
}
void next(node* u,int k,int &ans){
	if(u==NULL)return;
	if(u->v>k){
		if(u->v<ans)ans=u->v;
		if(u->son[0]!=NULL)next(u->son[0],k,ans);
	}
	else if(u->v==k&&u->flag>1){
		ans=u->v;
		return;
	}
	else if(u->v<=k)
	if(u->son[1]!=NULL)next(u->son[1],k,ans);
}
int main(){
	srand(987);
	int t,x,y;
	t=read();t--;
	answer=read();
	insert(rt,answer);
	while(t--){
		x=read();
		if(insert(rt,x))continue;
		int s1=-MAX,s2=MAX,p,q;
		front(rt,x,s1);next(rt,x,s2);
		answer+=min(abs(s1-x),abs(s2-x));
	}
	printf("%d\n",answer);
	return 0;
}