Splay Tree的C++实现

代码是参考网上神犇的..希望别被神犇怒D QAQ..

转载请注明出处，谢谢 http://blog.csdn.net/ACM_cxlove?viewmode=contents

1296 营业额统计

 

2002年

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

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

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

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

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

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

输入描述 Input Description

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

输出描述 Output Description

输出文件仅有一个正整数，即每天最小波动值之和，小于231

样例输入 Sample Input

6

5

1

2

5

4

6

样例输出 Sample Output

12

数据范围及提示 Data Size & Hint

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

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define N 100005
#define inf 1<<29
using namespace std;
int father[N],key[N],ch[N][2],root,tot1;  //分别表示父结点，键值，左右孩子(0为左孩子，1为右孩子),根结点，结点数量  
int n;
//新建一个结点  
void NewNode(int &r,int pre,int k) {
	r=++tot1;
	father[r]=pre;
	key[r]=k;
	ch[r][0]=ch[r][1]=0; //初始化左右孩子为空  
}
//旋转，kind为1为右旋，kind为0为左旋  
void Rotate(int x,int kind) {
	int y=father[x];
    //把其中一个分支先给父节点 
	ch[y][!kind]=ch[x][kind];
	father[ch[x][kind]]=y;
    //如果父节点不是根结点，则要和父节点的父节点连接起来  
	if(father[y])
		ch[father[y]][ch[father[y]][1]==y]=x;
	father[x]=father[y];
	ch[x][kind]=y;
	father[y]=x;
}
//Splay调整，将根为r的子树调整为goal  
void Splay(int r,int goal) {
	while(father[r]!=goal) {
    //父节点即是目标位置，goal为0表示，父节点就是根结点  
		if(father[father[r]]==goal)
			Rotate(r,ch[father[r]][0]==r);
		else {
			int y=father[r];
			int kind=ch[father[y]][0]==y;
            //两个方向不同，则先左旋再右旋  
			if(ch[y][kind]==r) {
				Rotate(r,!kind);
				Rotate(r,kind);
			}
            //两个方向相同，相同方向连续两次   
            else {
				Rotate(y,kind);
				Rotate(r,kind);
			}
		}
	}
    //更新根结点  
	if(goal==0) root=r;
}

int Insert(int k) {
	int r=root;
	while(ch[r][key[r]<k]) {
    //不重复插入 
		if(key[r]==k) {
			Splay(r,0);
			return 0;
		}
		r=ch[r][key[r]<k];
	}
	NewNode(ch[r][k>key[r]],r,k);
   //将新插入的结点更新至根结点  
	Splay(ch[r][k>key[r]],0);
	return 1;
}
//找前驱，即左子树的最右结点  
int get_pre(int x) {
	int tmp=ch[x][0];
	if(tmp==0)  return inf;
	while(ch[tmp][1])
		tmp=ch[tmp][1];
	return key[x]-key[tmp];
}
//找后继，即右子树的最左结点  
int get_next(int x) {
	int tmp=ch[x][1];
	if(tmp==0)  return inf;
	while(ch[tmp][0])
		tmp=ch[tmp][0];
	return key[tmp]-key[x];
}

int main() {
	scanf("%d",&n);
	root=tot1=0;
	int ans=0;
	for(int i=1; i<=n; i++) {
		int num;
		scanf("%d",&num);
		if(i==1) {
			ans+=num;
			NewNode(root,0,num);
			continue;
		}
		if(Insert(num)==0) continue;
		int a=get_next(root);
		int b=get_pre(root);
		ans+=min(a,b);
	}
	printf("%d\n",ans);
	return 0;
}