非旋转treap模板

来源:互联网 发布:深圳棋牌网络开发公司 编辑:程序博客网 时间:2024/06/05 18:33

先占坑


免责声明:本人由于同学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].rsint 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].fint 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);}