[平衡树-Splay]营业额统计

喜夏-厌秋 提交于 2020-03-06 00:25:04

题目

https://loj.ac/problem/10143

题解

  • 一眼就能看出这题就是找Splay的前驱和后继
  • 通过这题可以体会到为什么要加入一个无穷小的点和一个无穷大的点,加入可以防止出坑

代码

#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e5+7;
const int inf = 0x3f3f3f;
int rt;//根节点
int tot;//节点个数
struct node {
    int fa;//父亲节点
    int ch[2];//子节点
    int val;//权值
    int tag;//标记
    int sz;//子树大小
    int cnt;
}s[N];

struct Splay {
     int get(int x) {return s[s[x].fa].ch[1] == x;}

     void Clear(int x) {
         s[x].fa = s[x].ch[0] = s[x].ch[1] = s[x].sz = s[x].val = s[x].tag =0;
     }

     void maintain(int x){
         s[x].sz = s[s[x].ch[0]].sz + s[s[x].ch[1]].sz + s[x].cnt;
     }

     void Rorate(int x){
         int y = s[x].fa, z = s[y].fa, chk = get(x);

         s[y].ch[chk] = s[x].ch[chk ^ 1];
         s[s[x].ch[chk ^ 1]].fa = y;

         s[y].fa = x;
         s[x].ch[chk ^ 1] = y;

         s[x].fa =z;
         if(z) s[z].ch[s[z].ch[1] == y] = x;

         maintain(y);
         maintain(x);
     }

     void splay(int x,int y){
         for(int f = s[x].fa;f != y;Rorate(x),f=s[x].fa){
             if(s[f].fa != y) Rorate(get(x) == get(f) ? f : x);
         }
         if(y==0) rt = x;
     }

     void ins(int k){
         if(!rt) {
             s[++tot].val = k;
             s[tot].cnt++;
             rt = tot;
             maintain(rt);
             return ;
         }
         int now = rt,f = 0;
         while(true) {
             if(s[now].val == k) {
                 s[now].cnt++;
                 maintain(now);
                 maintain(f);
                 splay(now,0);
                 break;
             }
             f = now;
             now = s[now].ch[s[now].val < k];
             if(now == 0){
                 s[++tot].val = k;
                 s[tot].cnt++;
                 s[tot].fa = f;
                 s[f].ch[s[f].val < k] = tot;
                 maintain(tot);
                 maintain(f);
                 splay(tot,0);
                 break;
             }
         }
     }

     int getPre(){
         int now = s[rt].ch[0];
         while (s[now].ch[1]) now = s[now].ch[1];
         return now;
     }

     int getNxt(){
         int now = s[rt].ch[1];
         while(s[now].ch[0]) now = s[now].ch[0];
         return now;
     }
}st;
int main(){
    int n,x;
    scanf("%d",&n);
    st.ins(1000001);
    st.ins(-1000001);
    scanf("%d",&x);
    st.ins(x),n-=1;
    ll ans = x;
    while(n--){
        scanf("%d",&x);
        st.ins(x);
        if(s[rt].cnt > 1) continue;
        int aa = s[st.getPre()].val,bb = s[st.getNxt()].val,tmp = 1000001;
        if(aa != -1000001){
            tmp = abs(x - aa);
        }
        if(bb != 1000001){
            tmp = min(tmp,abs(x - bb));
        }
        if(tmp != 1000001) ans += tmp;
    }
    printf("%lld\n",ans);
    return 0;
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!