BZOJ 1588 营业额统计

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

平衡树裸题。

 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 using namespace std;
 #define maxn 33000
 int n,share[maxn];
 int lc[maxn],rc[maxn],father[maxn],root,ans;

 inline int small(int a,int b) {if (a < b) return a; return b;}

 inline void work();
 inline void zig(int);
 inline void zag(int);
 inline int find(int);
 inline void ins(int);
 inline void splay(int);

 int main()
 {
     freopen("1588.in","r",stdin);
     freopen("1588.out","w",stdout);
     scanf("%d",&n);
     int i;
     for (i = ;i<=n;i++)
         scanf("%d",share+i);
     work();
     printf("%d",ans);
     fclose(stdin); fclose(stdout);
     return ;
 }

 inline int find(int a)
 {
     int ret = ,now = root;
     ret = small(abs(share[now]-a),ret);
     now = rc[root];
     while (now > )
     {
         if (share[now] == a)
             return ;
         ret = small(abs(share[now]-a),ret);
         if (share[now] > a)
             now = rc[now];
         else now = lc[now];
     }
     now = lc[root];
     while (now > )
     {
         if (share[now] == a)
             return ;
         ret = small(abs(share[now]-a),ret);
         if (share[now] > a)
             now = rc[now];
         else now = lc[now];
     }
     return ret;
 }

 inline void work()
 {
     ans = share[]; root = ;
     int i,j,k;
     for (i = ;i<=n;i++)
     {
         k = find(share[i]);
         ans += k;
         if (k != )
             ins(i);
     }
 }

 inline void ins(int a)
 {
     int last,now;
     now = root;
     while (now != )
     {
         last = now;
         if (share[now] > share[a])
             now = rc[now];
         else now = lc[now];
     }
     father[a] = last;
     if (share[a] > share[last])
         lc[last] = a;
     else rc[last] = a;
     splay(a);
     root = a;
 }

 inline void zig(int x)
 {
     int y = father[x];
     father[x] = father[y];
     if (father[y] != )
     {
         if (lc[father[y]] == y)
             lc[father[y]] = x;
         else rc[father[y]] = x;
     }
     father[y] = x;
     lc[y] = rc[x];
     if (rc[x] != )
         father[rc[x]] = y;
     rc[x] = y;
 }

 inline void zag(int x)
 {
     int y = father[x];
     father[x] = father[y];
     if (father[y] != )
     {
         if (lc[father[y]] == y)
             lc[father[y]] = x;
         else rc[father[y]] = x;
     }
     father[y] = x;
     rc[y] = lc[x];
     if (lc[x] != )
         father[lc[x]] = y;
     lc[x] = y;
 }

 inline void splay(int a)
 {
     while (father[a] != )
     {
         if (father[father[a]] == )
         {
             if (a == lc[father[a]])
                 zig(a);
             else zag(a);
         }
         else
         {
             if (lc[father[father[a]]]==father[a] && lc[father[a]] == a)
             {
                 zig(father[a]),
                 zig(a);
             }
             else if (lc[father[father[a]]]==father[a] && rc[father[a]] == a)
             {
                 zag(a);
                 zig(a);
             }
             else if (rc[father[father[a]]]==father[a] && lc[father[a]] == a)
             {
                 zig(a);
                 zag(a);
             }
             else if (rc[father[father[a]]]==father[a] && rc[father[a]] == a)
             {
                 zag(father[a]);
                 zag(a);
             }
         }
     }
 }