【平衡树第一题庆祝】BZOJ1588HNOI2002营业额统计SBT做法

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

裸平衡树一道，插入节点然后找一下前驱后缀，稍微特判一下，最后就AC了

关键是这题数据有问题输入数据有空行卧槽。空行就默认按d=0处理。codevs还算友好自己出了数据无视这种问题，COGS和BZOJ各种WAWAWAWAWAWAWAWAWA呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵呵

河南省选我糊你熊脸

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXINT 100000000
using namespace std;
struct sbt
{
	int left,right,data,size;
}tree[41000];
int root,top;
int ans;
int n,d;
void left_rot(int &x)
{
	int y=tree[x].right;
	tree[x].right=tree[y].left;
	tree[y].left=x;
	tree[y].size=tree[x].size;
	tree[x].size=tree[tree[x].left].size+tree[tree[x].right].size+1;
	x=y;
}
void right_rot(int &x)
{
	int y=tree[x].left;
	tree[x].left=tree[y].right;
	tree[y].right=x;
	tree[y].size=tree[x].size;
	tree[x].size=tree[tree[x].left].size+tree[tree[x].right].size+1;
	x=y;
}
void maintain(int &x,bool flag)
{
	if (!flag)
	{
		if (tree[tree[tree[x].left].left].size>tree[tree[x].right].size)
		{
			right_rot(x);
		}
		else
		if (tree[tree[tree[x].left].right].size>tree[tree[x].right].size)
		{
			left_rot(tree[x].left);
			right_rot(x);
		}
		else
		return;
	}
	else
	{
		if (tree[tree[tree[x].right].right].size>tree[tree[x].left].size)
		{
			left_rot(x);
		}
		else
		if (tree[tree[tree[x].right].left].size>tree[tree[x].left].size)
		{
			right_rot(tree[x].right);
			left_rot(x);
		}
		else
		return;
	}
	maintain(tree[x].left,0);
	maintain(tree[x].right,1);
	maintain(x,1);
	maintain(x,0);
}
void insert(int &x,int data)
{
	if (x==0)
	{
		x=++top;
		tree[x].left=tree[x].right=0;
		tree[x].size=1;
		tree[x].data=data;
	}
	else
	{
		tree[x].size++;
		if (data<tree[x].data) insert(tree[x].left,data);
		else insert(tree[x].right,data);
		maintain(x,data>=tree[x].data);
	}
}
int prev(int x,int y,int data)
{
	if (x==0) return y;
	if (tree[x].data==data) return y;
	if (tree[x].data<data) return prev(tree[x].right,x,data);
	else return prev(tree[x].left,y,data);
}
int succ(int x,int y,int data)
{
	if (x==0) return y;
	if (tree[x].data==data) return x;
	if (tree[x].data>data) return succ(tree[x].left,x,data);
	else return succ(tree[x].right,y,data);
}
void print(int x)
{
	for (int i=1;i<=x;i++)
	{
		cout<<i<<" left"<<tree[i].left<<" right"<<tree[i].right<<" data"<<tree[i].data<<endl;
	}
}
int main()
{
	freopen("turnover.in","r",stdin);
	freopen("turnover.out","w",stdout);
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		if (scanf("%d",&d)==EOF) d=0;
		int a=succ(root,0,d);
		int b=prev(root,0,d);
		int tmpa=abs(d-tree[a].data);
		int tmpb=abs(d-tree[b].data);
		if (a==0) tmpa=MAXINT;
		if (b==0) tmpb=MAXINT;
		if (a||b) ans+=min(tmpa,tmpb);
		else ans+=d;
		insert(root,d);
		//print(i);
		//cout<<ans<<endl;
	}
	cout<<ans;
}