增加Split(分裂),Merge(合并)操作,非常好写,时间也不比普通treap慢什么。
#include<bits/stdc++.h>
using namespace std;
struct node
{
int l,r;int size,rnd,v;
}t[*];
int cnt,rt[];
void update(int k)
{
t[k].size=t[t[k].l].size+t[t[k].r].size+;
}
void newnode(int &k,int x)
{
t[k=++cnt].v=x;t[k].size=;t[k].rnd=rand();
}
int merge(int a,int b)
{
if(!a||!b)return a+b;
if(t[a].rnd>t[b].rnd)
{
int p=++cnt;t[p]=t[a];
t[p].r=merge(t[p].r,b);
update(p);
return p;
}
else
{
int p=++cnt;t[p]=t[b];
t[p].l=merge(a,t[p].l);
update(p);
return p;
}
}
void split(int now,int k,int &x,int &y)
{
if(!now)x=y=;
else
{
if(t[now].v<=k)
{
x=++cnt;t[x]=t[now];
split(t[x].r,k,t[x].r,y);
update(x);
}
else
{
y=++cnt;t[y]=t[now];
split(t[y].l,k,x,t[y].l);
update(y);
}
}
}
void Delete(int &root,int w)
{
int x=,y=,z=;
split(root,w,x,z);
split(x,w-,x,y);
y=merge(t[y].l,t[y].r);
root=merge(merge(x,y),z);
}
void Insert(int &root,int w)
{
int x=,y=,z=;
split(root,w,x,y);
newnode(z,w);
root=merge(merge(x,z),y);
}
int getval(int k,int w)
{
if(w==t[t[k].l].size+)return t[k].v;
else if(w<=t[t[k].l].size)return getval(t[k].l,w);
else return getval(t[k].r,w-t[t[k].l].size-);
}
int getkth(int &root,int w)
{
int x,y;
split(root,w-,x,y);
int ans=t[x].size+;
root=merge(x,y);
return ans;
}
int getpre(int &root,int w)
{
int x,y,k,ans;
split(root,w-,x,y);
if(!x)return -;
k=t[x].size;
ans=getval(x,k);
root=merge(x,y);
return ans;
}
int getnex(int &root,int w)
{
int x,y,ans;
split(root,w,x,y);
if(!y)return ;
else ans=getval(y,);
root=merge(x,y);
return ans;
}
int main()
{
int n,f,w,tim;
scanf("%d",&n);
for(int i=;i<=n;++i)
{
scanf("%d%d%d",&tim,&f,&w);
rt[i]=rt[tim];
if(f==)Insert(rt[i],w);
else if(f==)Delete(rt[i],w);
else if(f==)printf("%d\n",getkth(rt[i],w));
else if(f==)printf("%d\n",getval(rt[i],w));
else if(f==)printf("%d\n",getpre(rt[i],w));
else printf("%d\n",getnex(rt[i],w));
}
return ;
}