BZOJ 1588：营业额统计（Splay）

http://www.lydsy.com/JudgeOnline/problem.php?id=1588

题意：中文题意。

思路：每一个点每一个点插入Splay，然后插入新的一个点之后，查这个节点的前驱和后继，即左子树最右边的点和右子树最左边的点。然后再从两个点中取个差值较小的就是答案了。要注意Rotate的时候一些细节（要给 rt 的父亲的父亲更新其孩子的值），还有Splay的细节：如果 rt 和 父节点都是要旋转相同方向，应该先旋转父亲节点再旋 rt，如果旋转不同方向就都是旋 rt。

 #include <cstdio>
 #include <algorithm>
 #include <iostream>
 #include <cstring>
 #include <string>
 #include <cmath>
 #include <queue>
 #include <vector>
 using namespace std;
 #define INF 0x7fffffff
 #define N 40000
 struct node
 {
     int val, fa, son[]; // val是节点权值,fa是父亲,son[0]是左儿子, son[1]是右儿子
 }tree[N];
 int cnt; // 节点数

 void new_node(int f, int w, int kind)
 {
     cnt++;
     memset(tree[cnt].son, , sizeof(tree[cnt].son));
     tree[cnt].val = w;
     tree[cnt].fa = f;
     tree[f].son[kind] = cnt;
 }

 void Rotate(int rt, int kind) // 旋转操作要注意更新 rt 的父亲的父亲的儿子的值
 {
     int y = tree[rt].fa;
     tree[y].son[kind^] = tree[rt].son[kind];
     if(tree[rt].son[kind] != ) tree[tree[rt].son[kind]].fa = y;
     tree[rt].fa = tree[y].fa;
     int z = tree[rt].fa;
     if(tree[z].son[] == y) tree[z].son[] = rt; // 就是这里
     else tree[z].son[] = rt;
     tree[rt].son[kind] = y;
     tree[y].fa = rt;
 }

 void Splay(int rt, int goal)
 {
     if(tree[rt].fa == goal) return ;
     while(tree[rt].fa != goal) {
         int y = tree[rt].fa;
         int z = tree[y].fa;
         int kind1 = rt == tree[y].son[] ?  : ;
         int kind2 = y == tree[z].son[] ?  : ;
         if(z == goal) {
             Rotate(rt, kind1);
         } else {
             if(kind1 == kind2) { // 连续左旋或者右旋
                 Rotate(y, kind2);
             } else {
                 Rotate(rt, kind1); // 先左后右或者先右后左
             }
             Rotate(rt, kind2);
         }
     }
 }

 void Insert(int rt, int fa, int val, int kind)
 {
     if(rt == ) {
         new_node(fa, val, kind);
         return ;
     }
     if(tree[rt].val >= val) Insert(tree[rt].son[], rt, val, );
     else Insert(tree[rt].son[], rt, val, );
 }

 int Find(int rt, int kind)
 {
     if(tree[rt].son[kind] == ) return rt; // 查先驱和后继节点
     Find(tree[rt].son[kind], kind);
 }

 int main()
 {
     int n;
     while(~scanf("%d", &n)) {
         long long ans = ;
         cnt = ;
         for(int i = ; i <= n; i++) {
             int x;
             scanf("%d", &x);
             if(i == ) {
                 new_node(, x, );
                 ans += x;
             } else {
                 Insert(cnt, , x, );
                 Splay(cnt, );
                 int pre = , suf = ;
                 pre = Find(tree[cnt].son[], );
                 suf = Find(tree[cnt].son[], );
                 int prev = INF, sufv = INF;
                 if(pre != ) prev = tree[pre].val;
                 if(suf != ) sufv = tree[suf].val;
                 ans += min(abs(x - prev), abs(x - sufv));
             }
         }
         printf("%lld\n", ans);
     }
     return ;
 }