bzoj3224 Tyvj 1728 普通平衡树 非旋转式Treap

bzoj3224 Tyvj 1728 普通平衡树
裸平衡树
仔细想想 以前从来没徒手种过树真是羞耻
倒是学过些 但从来没有实现过导致平衡树的一些普通操作开始也是一脸懵逼
普通平衡树的基本操作:
rank(x,k):
k的子树下 <=x的最大值 的排名
如果当前节点u的data == x 走左子树 更新tmp tmp是刨除重复元素的答案
如果当前节点u的data < x 走右子树 更新答案
如果当前节点u的data > x 走左子树 不需要更新答案
xth(x,k)
k的子树下 排名第x的元素
u的排名 == x return
u的排名 > x 左走 u的排名 < x 右走
pre(x,k)//前驱
方法很多吧大概
u的data < x 走右子树 因为左子树的data小于等于u的data 不可能为前驱 更新答案
否则走左子树
neg(x,k)//后继
u的data > x 走左子树 因为右子树的data大于等于u的data 不可能为前驱 更新答案
否则走右子树
以上为平衡树的基本操作
非旋转的平衡树基于两个基本操作:
merge(a,b):返回一个treap,包含a,b两个treap中的所有节点。
split(a,n)返回两个treap l,r。
这个讲解的很详细嗯。
http://blog.csdn.net/zmh964685331/article/details/50536410

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<ctime>
#include<cstdlib>
#define inf (int)1e9
#define mp make_pair
#define pa pair<int,int>
using namespace std;
const int N=100005;
struct Node{int data,key,ls,rs,sz;}t[N];
int root=0,n,num=0;
void update(int x){
    t[x].sz=t[t[x].ls].sz+t[t[x].rs].sz+1;
}
int merge(int a,int b){// treap b中元素>=treap a中元素
    if(a==0 || b==0) return a+b;
    if(t[a].key<t[b].key) {t[a].rs=merge(t[a].rs,b);update(a);return a;}
    else {t[b].ls=merge(a,t[b].ls);update(b);return b;}
}
pa split(int a,int k){
    pa tmp;
    if(k==0) return mp(0,a);
    int ls=t[a].ls,rs=t[a].rs;
    if(t[t[a].ls].sz==k) {t[a].ls=0;update(a);return mp(ls,a);}
    if(t[t[a].ls].sz==k-1) {t[a].rs=0;update(a);return mp(a,rs);}
    if(t[t[a].ls].sz>k) {tmp=split(t[a].ls,k);t[a].ls=tmp.second;update(a);return mp(tmp.first,a);}
    if(t[t[a].ls].sz<k-1) {tmp=split(t[a].rs,k-t[t[a].ls].sz-1);t[a].rs=tmp.first;update(a);return mp(a,tmp.second);}
}
int Rank(int x,int k){
    int tmp=inf,ans=0;
    while(k){
        if(t[k].data==x) tmp=min(tmp,ans+t[t[k].ls].sz+1);
        if(t[k].data<x) ans+=t[t[k].ls].sz+1,k=t[k].rs;
        else k=t[k].ls;
    }
    return tmp==inf?ans:tmp;
}
void ins(int x){
    int k=Rank(x,root);
    t[++num].data=x;
    t[num].key=rand();
    t[num].sz=1;
    t[num].ls=t[num].rs=0;
    pa tmp=split(root,k);
    root=merge(tmp.first,num);
    root=merge(root,tmp.second);
}
void del(int x){
    int k=Rank(x,root);
    pa tmp1=split(root,k);
    pa tmp2=split(tmp1.first,k-1);
    root=merge(tmp2.first,tmp1.second);
}
int xth(int x,int k){
    while(1){
        if(t[t[k].ls].sz==x-1) return t[k].data;
        if(t[t[k].ls].sz<x-1) x-=(t[t[k].ls].sz+1),k=t[k].rs;
        else k=t[k].ls;
    }
}
int pre(int x,int k){
    int ans=-inf;
    while(k){
        if(t[k].data<x) ans=max(ans,t[k].data),k=t[k].rs;
        else k=t[k].ls;
    }
    return ans;
}
int neg(int x,int k){
    int ans=inf;
    while(k){
        if(t[k].data>x) ans=min(ans,t[k].data),k=t[k].ls;
        else k=t[k].rs;
    }
    return ans;
}
int main(){
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    srand(time(0));
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        int opt,x;
        scanf("%d%d",&opt,&x);
        if(opt==1) ins(x);
        if(opt==2) del(x);
        if(opt==3) printf("%d\n",Rank(x,root));//Rank(x) 返回 rank(x') x'为小于等于x的最大的数 
        if(opt==4) printf("%d\n",xth(x,root));
        if(opt==5) printf("%d\n",pre(x,root));
        if(opt==6) printf("%d\n",neg(x,root));
    }
    return 0;
}

你可能感兴趣的:(模板,treap,bzoj)