非旋转treap模板

先占坑


免责声明:本人由于同学zawedx神犇的带领喜欢缩♂行,若观看带来不适感,本人不负任何责任。
模板题bzoj 3224 普通平衡树
主要思(suo)想(hang)是把求value x在集合里面的rank的时候,视作求x-1的最小的后继(这样我们就能少打一个函数=w=)
pre和nxt都是pair类型,first和second分别代表的是前驱(后继)的rank和值
代码写的有点……难以看懂
看我不爽的可以去看看我的同学的代码,这样你看我的会舒服点(雾)

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
#define maxn 100100
#define L(x) t[x].ls
#define R(x) t[x].rs
int Q,opt,x,root,cnt,Re,F;
char ch;
struct pnt{int val,key,ls,rs,siz;}t[maxn];
typedef pair<int,int> P;
inline int read(){
    for(Re=F=0,ch=getchar();(ch<'0'||ch>'9')&&ch!='-';ch=getchar());
    if(ch=='-')F=1,ch=getchar();
    for(;'0'<=ch&&ch<='9';ch=getchar())Re=(Re<<3)+(Re<<1)+ch-'0';
    return F?-Re:Re;
}
inline int upd(int x){t[x].siz=t[t[x].ls].siz+t[t[x].rs].siz+1;return x;}
int merge(int x,int y){
    if(!x||!y)return x|y;
    if(t[x].key<t[y].key)return R(x)=merge(R(x),y),upd(x);
    else return L(y)=merge(x,L(y)),upd(y);
}
P split(int x,int k){
    if(k==0)return P(0,x);
    int l=t[x].ls,r=t[x].rs;
    if(t[l].siz==k)return L(x)=0,P(l,upd(x));
    if(t[l].siz+1==k)return R(x)=0,P(upd(x),r);
    if(k<t[L(x)].siz){
        P tmp=split(l,k);
        return L(x)=tmp.second,P(tmp.first,upd(x));
    }
    P tmp=split(r,k-t[l].siz-1);
    return R(x)=tmp.first,P(upd(x),tmp.second);
}
inline P nxt(int x){
    P tmp(1,2047483647);
    for(int now=root;now;)
        if(t[now].val<=x)tmp.first+=t[L(now)].siz+1,now=R(now);
        else tmp.second=min(t[now].val,tmp.second),now=L(now);
    return tmp;
}
inline P pre(int x){
    P tmp(t[root].siz,-2047483647);
    for(int now=root;now;)
        if(t[now].val>=x)tmp.first-=t[R(now)].siz+1,now=L(now);
        else tmp.second=max(t[now].val,tmp.second),now=R(now);
    return tmp;
}
inline void ins(int x){
    P tmp=split(root,nxt(x-1).first-1);
    t[++cnt]=(pnt){x,rand(),0,0,1};
    root=merge(merge(tmp.first,cnt),tmp.second);
}
inline void del(int x){
    P tmp=split(root,nxt(x-1).first-1);
    root=merge(tmp.first,split(tmp.second,1).second);
}
int num(int x){
    for(int now=root;now;){ 
        if(t[L(now)].siz==x-1)return t[now].val;
        else if(t[L(now)].siz>=x)now=L(now);
        else x-=t[L(now)].siz+1,now=R(now);
    } 
}
int main(){
    srand(20010426);
    for(Q=read();Q--;){
        switch(read()){
        case 1:ins(read());break;
        case 2:del(read());break;
        case 3:printf("%d\n",nxt(read()-1).first);break;
        case 4:printf("%d\n",num(read()));break;
        case 5:printf("%d\n",pre(read()).second);break;
        case 6:printf("%d\n",nxt(read()).second);break;
        }
    }
}

模板题2 bzoj 3223 文艺平衡树 容易发现翻转操作对某一整块翻转偶数次之后,只改变其位置而不改变其内部元素顺序。故对每个节点增加一个Flag标记,用^1维护就好了。

#include<cstdio>
#include<algorithm>
#include<cstdlib>
using namespace std;
#define maxn 100100
#define P pair<int,int>
#define L(x) t[x].ls
#define R(x) t[x].rs
#define F(x) t[x].f
int l,r,m,n,cnt,root;
struct pnt{int val,key,S,ls,rs;bool f;}t[maxn];
inline int U(int x){t[x].S=t[L(x)].S+t[R(x)].S+1;return x;}
inline int ps(int x){if(F(x))swap(L(x),R(x)),F(L(x))^=1,F(R(x))^=1,F(x)^=1;return x;}
void get(int &a,int &b,P pp){a=pp.first,b=pp.second;}
int meg(int x,int y){
    return !x||!y?x|y:t[x].key<t[y].key?
    (R(ps(x))=meg(R(x),y),U(x)):
    (L(ps(y))=meg(x,L(y)),U(y));
}
P spl(int x,int k){
    if(!k)return P(0,x);
    int T=k-t[L(ps(x))].S-1;
    return T>=0?(get(R(x),T,spl(R(x),T)),P(U(x),T)):
    (get(T,L(x),spl(L(x),k)),P(T,U(x)));
}
void prf(int x){
    if(t[ps(x)].S==1)printf("%d ",t[x].val);
    else if(x)prf(L(x)),printf("%d ",t[x].val),prf(R(x));
}
int main(){
    scanf("%d%d",&n,&m);
    srand(20010426);
    for(int i=1;i<=n;i++)t[++cnt]=(pnt){i,rand(),1,0,0},root=meg(root,cnt);
    while(m--){
        scanf("%d%d",&l,&r);
        P tmp=spl(root,l-1);
        P lsh=spl(tmp.second,r-l+1);
        F(lsh.first)^=1;
        root=meg(meg(tmp.first,lsh.first),lsh.second);
    }
    prf(root);
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值