两个一起学的,就放一块了。
主要是用来存板子。
//This is a Splay Tree.
#include <cstdio>
#include <cstring>
using namespace std;
const int N=1e5+5,INF=0x3f3f3f3f;
int n,root,cntnode;
struct node //始终满足左小右大
{
int fa,ch[2],val,siz,cnt;
//int mark; //区间反转标记
}t[N];
inline bool get(int x) {return t[t[x].fa].ch[1]==x;} //右儿子?1:0
inline void upd(int x) {t[x].siz=t[t[x].ch[0]].siz+t[t[x].ch[1]].siz+t[x].cnt;} //更新计数
inline void zigzag(int x) //旋转操作
{
int fa=t[x].fa; int gfa=t[fa].fa;
int d1=get(x),d2=get(fa);
t[fa].ch[d1]=t[x].ch[d1^1]; t[t[x].ch[d1^1]].fa=fa; //断开fa与x,连接fa与x的儿子
t[gfa].ch[d2]=x; t[x].fa=gfa; //断开gfa与fa,连接gfa与x
t[fa].fa=x; t[x].ch[d1^1]=fa; //连接x与fa
upd(fa); upd(x); //一定是先upd fa再upd x,因为此时fa是x的节点
}
void splay(int x,int goal) //伸展操作,即将一个节点x不断旋转至goal的儿子的位置
{
while(t[x].fa!=goal)
{
int fa=t[x].fa; int gfa=t[fa].fa;
int d1=get(x),d2=get(fa);
if(gfa!=goal)
{
if(d1==d2) zigzag(fa); //双旋操作——如果x与fa处在同一方向,要先旋转fa
else zigzag(x);
}
zigzag(x);
}
if(goal==0) root=x; //这里用goal=0来实现把x变为根节点的操作
}
//以上为维持splayTree功能的基本操作,以下将实现splay的几种常见用途
void insert(int val) //插入值功能
{
int node=root,fa=0;
while(node && t[node].val!=val)
fa=node,node=t[node].ch[t[node].val<val]; //通过不断遍历树来找到插入位置
if(node) t[node].cnt++; //已有此编号
else
{
node=++cntnode; if(fa) t[fa].ch[t[fa].val<val]=node;
t[node].fa=fa; t[node].val=val; t[node].cnt=t[node].siz=1;
}
splay(node,0);
}
int kth(int k) //第k小的数
{
int node=root;
for(;;)
{
int son=t[node].ch[0];
if(k<=t[son].siz) node=son;
else if(k>t[son].siz+t[node].cnt)
k-=t[son].siz+t[node].cnt,
node=t[node].ch[1];
else return t[node].val;
}
}
int find(int val) //查找值
{
int node=root;
while(t[node].val!=val && t[node].ch[t[node].val<val])
node=t[node].ch[t[node].val<val];
return node;
}
int getrk(int val)
{
splay(find(val),0);
return t[t[root].ch[0]].siz;
}
int presuc(int val,int tp) //查找前驱后继,tp=0为前驱,tp=1为后继
{
splay(find(val),0); int node=root;
if(t[node].val<val&&!tp || t[node].val>val&&tp) //如果找到的节点满足要求就直接返回
return node;
node=t[node].ch[tp];
while(t[node].ch[tp^1])
node=t[node].ch[tp^1]; //否则找节点左/右子树中最靠右/左的节点(根据平衡树的性质
return node;
}
void delet(int val)
{
int pre=presuc(val,0),suc=presuc(val,1);
splay(pre,0); splay(suc,pre); //将pre旋转到根,suc旋转到pre的下面,那么suc的左子树就是要删除的
if(t[t[suc].ch[0]].cnt>1)
{
t[t[suc].ch[0]].cnt--; //常规的删除
splay(t[suc].ch[0],0);
}
else t[suc].ch[0]=0;
}
int main()
{
insert(-INF); insert(INF);
scanf("%d",&n);
for(int i=1,opt,x;i<=n;++i)
{
scanf("%d%d",&opt,&x);
if(opt==1) insert(x);
if(opt==2) delet(x);
if(opt==3) printf("%d\n",getrk(x));
if(opt==4) printf("%d\n",kth(x+1));
if(opt==5) printf("%d\n",t[presuc(x,0)].val);
if(opt==6) printf("%d\n",t[presuc(x,1)].val);
}
return 0;
}
#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=100010;
int tot,root;
struct FHQtreap
{
int ch[2],val,pri,siz;
#define l(x) t[x].ch[0]
#define r(x) t[x].ch[1]
}t[N];
inline void upd(int x) {t[x].siz=t[l(x)].siz+t[r(x)].siz+1;}
inline int newNode(int x)
{
t[++tot].siz=1,t[tot].val=x,t[tot].pri=rand();
return tot;
}
int merge(int x,int y)
{
if(!x||!y) return x|y;
if(t[x].pri<t[y].pri) return r(x)=merge(r(x),y),upd(x),x;
else return l(y)=merge(x,l(y)),upd(y),y;
}
void split(int now,int k,int &x,int &y)
{
if(!now) {x=y=0; return;}
if(t[now].val<=k) x=now,split(r(now),k,r(now),y);
else y=now,split(l(now),k,x,l(now));
upd(now);
}
int kth(int now,int k)
{
while(1)
if(k<=t[l(now)].siz) now=l(now);
else
{
if(k==t[l(now)].siz+1) return now;
else k-=t[l(now)].siz+1,now=r(now);
}
}
int main()
{
srand(19260817);
int T; scanf("%d",&T);
while(T--)
{
int opt,a,x,y,z;
scanf("%d%d",&opt,&a);
if(opt==1)
{
split(root,a,x,y);
root=merge(merge(x,newNode(a)),y);
}
if(opt==2)
{
split(root,a,x,z);
split(x,a-1,x,y);
y=merge(l(y),r(y));
root=merge(merge(x,y),z);
}
if(opt==3)
{
split(root,a-1,x,y);
printf("%d\n",t[x].siz+1);
root=merge(x,y);
}
if(opt==4) printf("%d\n",t[kth(root,a)].val);
if(opt==5)
{
split(root,a-1,x,y);
printf("%d\n",t[kth(x,t[x].siz)].val);
root=merge(x,y);
}
if(opt==6)
{
split(root,a,x,y);
printf("%d\n",t[kth(y,1)].val);
root=merge(x,y);
}
}
return 0;
}
手机扫一扫
移动阅读更方便
你可能感兴趣的文章