码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-03-06 01:46:50      阅读:71      评论:0      收藏:0      [点我收藏+]

标签:its   problem   names   平衡   using   父亲节   fas   str   for   

题目

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;
}

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

标签:its   problem   names   平衡   using   父亲节   fas   str   for   

原文地址:https://www.cnblogs.com/smallocean/p/12423984.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!