您的位置:首页 > 其它

非旋转treap模板

2017-01-24 11:35 375 查看
先挑个好讲一点的试试,原理什么的我最多一句带过就算了

你问我无旋treap的原理?我只能回答你一句无可奉告

我学这个东西的时候,想找别人的代码看看

看到别人写得那么长,就自己写了下

然后,就写成现在这样了

----------------------------------------------------------------------------------

#define mp make_pair
#define L a[t].l
#define R a[t].r
#define F fa[t]
typedef pair<int,int> par;
首先是merge,返回值为合并后的根

若其中一个为空树直接返回,否则递归

那么,很自然两个if就没了,这里用?:也没什么不吼对吧

int mer(int x,int y){
return x==0||y==0?x+y:a[y].k>a[x].k?
(a[x].r=mer(a[x].r,y),up(x),x):
(a[y].l=mer(x,a[y].l),up(y),y);
}
然后是split,按个数split用递归,按编号split用自底向上(比如维护括号序列,确定了代表这个括号的node)

递归的split,返回值是两个树的根组成的pair

若无需分裂当前树,即k==sum||k==0时,直接返回

否则用sz判下切哪边,然后更新,返回

什么?还要开个tmp记录返回值?为了方便可以写成这样

void get(int &x,int &y,par t){x=t.first,y=t.second;}
其他操作也会用到这东西,可以看看下面那道模板题里面用的get

除此之外,经过尝试,我发现在这开个int写起来是最舒服的

par spl(int t,int k){
if (!k) return mp(t,0);
if (k==a[t].sz) return mp(0,t);
int p=k-a[R].sz-1; return p>=0?
(get(p,L,spl(L,p)),up(t),mp(p,t)):
(get(R,p,spl(R,k)),up(t),mp(t,p));
}
自底向上split(其他什么build,rank,insert,delete之类的不管了

这个第一次写是在考场上,就没怎么缩,要多维护个fa数组,split的时候要清fa

par spl(int t){
int l=L,r=R,tp=t;
for (;F;t=F) a[F].l==t?
r=fa[a[F].l=r]=F:
l=fa[a[F].r=l]=F;
t=tp,L=R=F=fa[l]=fa[r]=0;
return mp(l,r);
}
就这样,无旋treap写完啦~\(≧▽≦)/~

------------------------------------------------------------------------------

模板题:CV1343蚱蜢

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
#define N 100010
#define mp make_pair
#define L a[t].l
#define R a[t].r
typedef pair<int,int> par;
struct data{int l,r,v,mx,sz,k;}a
;
int up(int t){
a[t].sz=a[L].sz+a[R].sz+1;
a[t].mx=max(a[t].v,max(a[L].mx,a[R].mx));
}
int mer(int x,int y){ return x==0||y==0?x+y:a[y].k>a[x].k? (a[x].r=mer(a[x].r,y),up(x),x): (a[y].l=mer(x,a[y].l),up(y),y); }
int get(int &x,int &y,par t){x=t.first,y=t.second;}
par spl(int t,int k){ if (!k) return mp(t,0); if (k==a[t].sz) return mp(0,t); int p=k-a[R].sz-1; return p>=0? (get(p,L,spl(L,p)),up(t),mp(p,t)): (get(R,p,spl(R,k)),up(t),mp(t,p)); }
int n,m,i,r,r1,r2,x,y;
char ch;
inline int rd(){int t;scanf("%d",&t);return t;}
int main(){
srand(233);
n=rd(),m=rd();
for (i=1;i<=n;i++) a[i].mx=a[i].v=rd(),
a[i].k=rand(),a[i].sz=1;
for (r=1,i=2;i<=n;i++) r=mer(r,i);
while (m--){
x=rd();while ((ch=getchar())!='D'&&ch!='L');y=rd();
if (ch=='D'){
get(r,r1,spl(r,n-x));
get(r1,r2,spl(r1,n-x-y));
cout<<a[r1].mx<<"\n";
get(r,r2,spl(mer(r,r2),n-x-y+1));
r=mer(r=mer(r,r1),r2);
}
else {
get(r,r2,spl(r,n-x+1));
get(r,r1,spl(r,y));
cout<<a[r1].mx<<"\n";
get(r,r2,spl(mer(r,r2),n-x));
r=mer(r=mer(r,r1),r2);
}
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  非旋转treap 模板