wikioi1296营业额统计题解

• 题目描述 Description
  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

• Solution
  平衡树模板题。每次读取ai，将其插入平衡树中，然后查其前趋和后继，取差绝对值较小的加入答案。练了练SBT，但发现有一个问题不好解决。当SBT返回一个与ai相同的值时，究竟是前面有一天营业额恰好是ai还是ai没有前趋或者后继呢？我用了一个数组记录元素出现的频数。因为有负数，还要整体平移。

• Code

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 2000000;
int s[maxn], lc[maxn], rc[maxn], key[maxn];
int n, ans, tot;
int root = 1;
int flag[10000001], delta = 2000000;
inline void read(int &a)
{
    int f = 1;
    char ch = getchar();
    a = 0;
    while(ch < '0' || ch > '9')
    {
        if(ch == '-') f = -1;
        ch = getchar();
    }
    while(ch >= '0' && ch <= '9')
    {
        a = a*10 + ch - 48;
        ch = getchar();
    }
    a *= f;
}
inline void write(int a)
{
    int top = 0;
    char ch[30];
    if(a < 0)
    {
        putchar('-');
        a = -a;
    }
    do {
        ch[top++] = a%10 + 48;
        a /= 10;
    } while(a);
    while(top--) putchar(ch[top]);
    putchar('\n');
}
inline void rr(int &t)
{
    int k = lc[t];
    lc[t] = rc[k];
    rc[k] = t;
    s[k] = s[t];
    s[t] = s[lc[t]] + s[rc[t]] + 1;
    t = k;
}
inline void lr(int &t)
{
    int k = rc[t];
    rc[t] = lc[k];
    lc[k] = t;
    s[k] = s[t];
    s[t] = s[lc[t]] + s[rc[t]] + 1;
    t = k;
}
void maintain(int &t, bool flag)
{
    if(!flag)
    {
        if(s[lc[lc[t]]] > s[rc[t]]) rr(t);
        else if(s[rc[lc[t]]] > s[rc[t]])
        {
            lr(lc[t]); rr(t);
        }
        else return;
    }
    else
    {
        if(s[rc[rc[t]]] > s[lc[t]]) lr(t);
        else if(s[lc[rc[t]]] > s[lc[t]])
        {
            rr(rc[t]); lr(t);
        }
        else return;
    }
    maintain(lc[t], false);
    maintain(rc[t], true);
    maintain(t, true);
    maintain(t, false);
}
void insert(int &t, int v)
{
    if(t == 0) {
        key[t = ++tot] = v;
        s[t] = 1;
        lc[t] = rc[t] = 0;
    }
    else
    {
        ++s[t];
        if(v < key[t]) insert(lc[t], v);
        else insert(rc[t], v);
        maintain(t, v >= key[t]);
    }
}
int dlete(int &t, int v)
{
    --s[t];
    int ans;
    if(v == key[t] || (v < key[t] && lc[t] == 0) || (v > key[t] && rc[t] == 0))
    {
        ans = key[t];
        if(lc[t] == 0 || rc[t] == 0) t = lc[t] + rc[t];
        else key[t] = dlete(lc[t], key[t] + 1);
    }
    else if(v < key[t]) ans = dlete(lc[t], v);
    else ans = dlete(rc[t], v);
    return ans;
}
inline bool find(int t, int v)
{
    while(t != 0)
    {
        if(v == key[t]) return true;
        else if(v < key[t]) t = lc[t];
        else t = rc[t];
    }
    return false;
}
inline int select(int t, int k)
{
    while(k != s[lc[t]] + 1)
    {
        if(k < s[lc[t]] + 1) t = lc[t];
        else { k -= (s[lc[t]] + 1); t = rc[t]; }
    }
    return key[t];
}
inline int rank(int t, int v)
{
    int ans = 1;
    while(t != 0)
    {
        if(v <= key[t]) t = lc[t];
        else { ans += s[lc[t]] + 1; t =rc[t]; }
    }
    return ans;
}
int pred(int t, int v)
{
    if(t == 0) return v;
    if(v <= key[t]) return pred(lc[t], v);
    else
    {
        int d = pred(rc[t], v);
        if(d == v) d = key[t];
        return d;
    }
}
int succ(int t, int v)
{
    if(t == 0) return v;
    if(v >= key[t]) return succ(rc[t], v);
    else
    {
        int d = succ(lc[t], v);
        if(d == v) d = key[t];
        return d;
    }
}
int main()
{
    int a, b, c;
    read(n);
    for(int i = 1; i <= n; ++i)
    {
        read(a);
        ++flag[a+delta];
        insert(root, a);
        if(flag[a+delta] > 1) continue;
        b = pred(root, a); c = succ(root, a);
        if(b == a)
        {
            if(c == a) 
                if(tot == 1) ans += a; else;
            else ans += c - a;
        }
        else
        {
            if(c == a) ans += a - b;
            else ans += min(a - b, c - a);
        }
    }
    write(ans);
    return 0;
}