bzoj #1588 营业额统计（splay）

题目链接

Description

Tiger最近被公司升任为营业部经理，他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况。 Tiger拿出了公司的账本，账本上记录了公司成立以来每天的营业额。分析营业情况是一项相当复杂的工作。由于节假日，大减价或者是其他情况的时候，营业额会出现一定的波动，当然一定的波动是能够接受的，但是在某些时候营业额突变得很高或是很低，这就证明公司此时的经营状况出现了问题。经济管理学上定义了一种最小波动值来衡量这种情况： 该天的最小波动值 当最小波动值越大时，就说明营业情况越不稳定。 而分析整个公司的从成立到现在营业情况是否稳定，只需要把每一天的最小波动值加起来就可以了。你的任务就是编写一个程序帮助Tiger来计算这一个值。 第一天的最小波动值为第一天的营业额。  输入输出要求

Input

第一行为正整数 ，表示该公司从成立一直到现在的天数，接下来的n行每行有一个整数(有可能有负数) ，表示第i
天公司的营业额。
天数n<=32767,
每天的营业额ai <= 1,000,000。
最后结果T<=2^31

Output

输出文件仅有一个正整数，即Sigma(每天最小的波动值) 。结果小于2^31 。

---

一道裸的splay题。
题目大意时每次插入一个数，同时寻找与当前数差值最小的数。
随机数据可以用二叉树直接过，但是如果所有的数是单调递减的，那么复杂度就退化到了O(n^2)。
考虑splay，每次插入一个数splay后，再BST内寻找前驱和后继。
时间复杂度保证为O(nlogn)。

P.S:代码中rotate和splay的压(zhuang)行(bi)写法不是很推荐，因为比赛时这样写很容易写崩。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define maxn 100050
#define INF 1000000000
using namespace std;
typedef long long ll;
struct node
{
    int f,s[2],w; //father,leftson,rightson
}a[maxn*2];
int dfn=0,root,y,z,bo,n,x,p,q,hz;
ll ans=0ll;
inline void rotate(int x,int k) //k=0左旋,k=1右旋 
{
    y=a[x].f,z=a[y].f,a[y].s[!k]=a[x].s[k];
    if (a[x].s[k]!=-1) a[a[x].s[k]].f=y;
    if (z!=-1) a[z].s[a[z].s[1]==y]=x;
    a[x].f=z,a[y].f=x,a[x].s[k]=y;
} 
inline void splay(int x)
{
    while (a[x].f!=-1)
    {
        int y=a[x].f,z=a[y].f;
        if (z==-1) rotate(x,a[y].s[0]==x);
        else
        {
            if (a[z].s[0]==y) {if (a[y].s[0]==x) rotate(y,1),rotate(x,1);else rotate(x,0),rotate(x,1);}
            else {if (a[y].s[1]==x) rotate(y,0),rotate(x,0);else rotate(x,1),rotate(x,0);}
        }
    }
    root=x; 
}
inline int pre(int x)   //寻找前驱
{
    int tmp=a[x].s[0];
    if (tmp==-1) return tmp;
    while (a[tmp].s[1]!=-1) tmp=a[tmp].s[1];
    return tmp;
} 
inline int suc(int x)  //寻找后继 
{
    int tmp=a[x].s[1];
    if (tmp==-1) return tmp;
    while (a[tmp].s[0]!=-1) tmp=a[tmp].s[0];
    return tmp;
}
inline void insert(int now,int x)
{
    if (a[x].w==now) {bo=0,splay(x);return;}   //已有相同元素的节点
    if (now<a[x].w)
    {
        if (a[x].s[0]==-1) a[x].s[0]=dfn,a[dfn].f=x,a[dfn].s[0]=a[dfn].s[1]=-1,a[dfn].w=now; 
        else insert(now,a[x].s[0]);
    }
    else
    {
        if (a[x].s[1]==-1) a[x].s[1]=dfn,a[dfn].f=x,a[dfn].s[0]=a[dfn].s[1]=-1,a[dfn].w=now; 
        else insert(now,a[x].s[1]);
    } 
}
int main()
{
    scanf("%d",&n);
    scanf("%d",&x);
    ans=x,dfn++,a[dfn].f=a[dfn].s[0]=a[dfn].s[1]=-1,a[dfn].w=x,root=dfn;  //建立根节点 
    for (int i=1;i<n;i++) 
    {
        scanf("%d",&x);
        bo=1,dfn++,insert(x,root);
        if (!bo) continue;  //已有过相同元素的节点，那么该点对答案肯定没有贡献 
        splay(dfn),p=pre(dfn),q=suc(dfn),hz=INF;
        if (p!=-1) hz=min(hz,abs(x-a[p].w));
        if (q!=-1) hz=min(hz,abs(x-a[q].w)); 
        ans+=hz;   //比较x与其前驱与后继的距离后累加答案 
    }
    printf("%lld",ans);
}