首页 > 其他 > 详细

洛谷——P3369 【模板】普通平衡树

时间:2018-10-10 23:26:42      阅读:181      评论:0      收藏:0      [点我收藏+]

P3369 【模板】普通平衡树

 

平衡树大法好,蒟蒻(博主)最近正在收集高级数据结构的碎片,企图合成数据结构的元素之力来使自己的RP++。。。

 

您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:

  1. 插入xx数
  2. 删除xx数(若有多个相同的数,因只删除一个)
  3. 查询xx数的排名(排名定义为比当前数小的数的个数+1+1。若有多个相同的数,因输出最小的排名)
  4. 查询排名为xx的数
  5. xx的前驱(前驱定义为小于xx,且最大的数)
  6. xx的后继(后继定义为大于xx,且最小的数)

 

教程地址

 

博主太蒟,只能抄模板,曾经抄了个假的模板,有点=_=

 

#include<iostream>
#include<cstdio>

#define N 10000000
using namespace std;

int ch[N][2],cnt[N],val[N],size[N],par[N],root,ncnt;

int chk(int x){
    return ch[par[x]][1]==x;
}

void pushup(int x){
    size[x]=size[ch[x][0]]+size[ch[x][1]]+cnt[x];
}

void rotate(int x){
    int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
    ch[y][k]=w,par[w]=y;//父亲 
    ch[z][chk(y)]=x,par[x]=z;//祖父 
    ch[x][k^1]=y,par[y]=x;//自己 
    pushup(y),pushup(x);
}

void splay(int x,int goal=0){
    while(par[x]!=goal){
        int y=par[x],z=par[y];
        if(z!=goal){
            if(chk(y)==chk(x)) rotate(y);
            else rotate(x);
        }
        rotate(x);
    }
    if(!goal) root=x;
}

void find(int x){
//    if(!root) return;
    int cur=root;
    while(ch[cur][x>val[cur]]&&x!=val[cur]){
        cur=ch[cur][x>val[cur]];
    }
    splay(cur);
}

void insert(int x){
    int cur=root,p=0;
    while(cur&&val[cur]!=x){
        p=cur;
        cur=ch[cur][x>val[cur]];
    }
    if(cur){
        cnt[cur]++;
    }else{
        cur=++ncnt;
        if(p) ch[p][x>val[p]]=cur;
        ch[cur][0]=ch[cur][1]=0;
        val[cur]=x,par[cur]=p;
        cnt[cur]=size[cur]=1;
    }
    splay(cur);
}

int kth(int k){
    int cur=root;
    while(1){
        if(ch[cur][0]&&k<=size[ch[cur][0]]){
            cur=ch[cur][0];
        }else if(k>size[ch[cur][0]]+cnt[cur]){
            k-=size[ch[cur][0]]+cnt[cur];
            cur=ch[cur][1];
        }else return cur;
    }
}

int pre(int x){
    find(x);
    if(val[root]<x) return root;
    int cur=ch[root][0];
    while(ch[cur][1]){
        cur=ch[cur][1];
    }
    return cur;
}
int succ(int x){
    find(x);
    if(val[root]>x) return root;
    int cur=ch[root][1];
    while(ch[cur][0]){
        cur=ch[cur][0];
    }
    return cur;
}

void remove(int x){
    int last=pre(x),next=succ(x);
    splay(last);splay(next,last);
    int del=ch[next][0];
    if(cnt[del]>1){
        cnt[del]--;
        splay(del);
    }else ch[next][0]=0;
}

int n;

int main()
{
    scanf("%d",&n);
    insert(0x3f3f3f3f);
    insert(0xcfcfcfcf);
    for(int opt,x,i=1;i<=n;i++){
        scanf("%d%d",&opt,&x);
        if(opt==1) insert(x);
        if(opt==2) remove(x);
        if(opt==3) find(x),printf("%d\n",size[ch[root][0]]);
        if(opt==4) printf("%d\n",val[kth(x+1)]);
        if(opt==5) printf("%d\n",val[pre(x)]);
        if(opt==6) printf("%d\n",val[succ(x)]);
    }
    
    return 0;
}

 

洛谷——P3369 【模板】普通平衡树

原文:https://www.cnblogs.com/song-/p/9769290.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!