【BZOJ-1588】营业额统计 Splay

1588: [HNOI2002]营业额统计

Time Limit: 5 Sec  Memory Limit: 162 MB
Submit: 12485  Solved: 4508
[Submit][Status][Discuss]

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

此题数据有问题，详见讨论版http://www.lydsy.com/JudgeOnline/wttl/wttl.php?pid=1588

Source

Solution

Splay模板题，读一个x，插入x，然后查x的前驱后继，分类讨论一下....

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0' && ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
#define maxn 100010
int n;
int sz,root;
int fa[maxn],son[maxn][2],key[maxn],cnt[maxn],size[maxn];
void clear(int now)
{
    son[now][1]=son[now][0]=fa[now]=key[now]=size[now]=cnt[now]=0;
}
int get(int now)
{
    return son[fa[now]][1]==now;
}
void update(int now)
{
    if (!now) return;
    size[now]=cnt[now];
    if (son[now][0]) size[now]+=size[son[now][0]];
    if (son[now][1]) size[now]+=size[son[now][1]];
}
void rotate(int now)
{
    int old=fa[now],oldf=fa[old],which=get(now);
    son[old][which]=son[now][which^1]; fa[son[old][which]]=old;
    fa[old]=now; son[now][which^1]=old; fa[now]=oldf;
    if (oldf) son[oldf][son[oldf][1]==old]=now;
    update(old); update(now);
}
void splay(int now)
{
    for (int f; (f=fa[now]); rotate(now))
        if (fa[f]) 
            if (get(now)==get(f)) 
                rotate(f); else rotate(now);
    root=now;
}
void insert(int v)
{
    if (!root) 
        {sz++;son[sz][0]=son[sz][1]=fa[sz]=0;key[sz]=v;size[sz]=1;cnt[sz]=1;root=sz;return;}
    int now=root,f=0;
    while (true)
        {
            if (key[now]==v)
                {cnt[now]++;update(now);update(f);splay(now);break;}
            f=now; now=son[now][key[now]<v];
            if (now==0)
                {
                    sz++;son[sz][0]=son[sz][1]=0;key[sz]=v;size[sz]=1;
                    cnt[sz]=1;fa[sz]=f;son[f][key[f]<v]=sz;
                    update(f);splay(sz);break;
                }
        }
}
int prev()
{
    int now=son[root][0];
    while (son[now][1]) now=son[now][1];
    return now;
}
int succ()
{
    int now=son[root][1];
    while (son[now][0]) now=son[now][0];
    return now;
} 
int main()
{
    n=read(); int ans=0;
    for (int i=1; i<=n; i++)
        {
            int x=read(); insert(x);
            int pre=key[prev()],suc=key[succ()];
            if (i==1) {ans=x;continue;}
            if (cnt[root]>1) continue;
            if (prev() && succ())
                if (abs(pre-x)<abs(suc-x)) ans+=abs(pre-x);
                    else ans+=abs(suc-x);
            else if (!prev()) ans+=abs(suc-x);
            else if (!succ()) ans+=abs(pre-x);
        }
    printf("%d\n",ans);
    return 0;
}

一开始没看见BZOJ有问题..鬼畜不止..还有做这种水题，一开始9A1W，而且就差1，感觉很奇妙，发现手误打错一个变量，MD这都能过九个不报错...