1. 程式人生 > >非旋轉treap模板

非旋轉treap模板

先挑個好講一點的試試,原理什麼的我最多一句帶過就算了
你問我無旋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[N];
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;
}