poj 3481 Double Queue(平衡树，SBT)

poj 3481 Double Queue

平衡树模板题，练手

#include<stdio.h>
#include<string.h>
#define MAXN 100005
struct node
{
int l,r;
int key,id;
int size;
}t[MAXN];
int top,root;
void lRotate(int &p)
{
int rch=t[p].r;
t[p].r=t[rch].l;
t[rch].l=p;

t[rch].size=t[p].size;
t[p].size=t[t[p].l].size+t[t[p].r].size+1;

p=rch;
}
void rRotate(int &p)
{
int lch=t[p].l;
t[p].l=t[lch].r;
t[lch].r=p;

t[lch].size=t[p].size;
t[p].size=t[t[p].l].size+t[t[p].r].size+1;

p=lch;
}
void maintain(int &p,bool flag)
{
if(!flag)
{
if(t[t[t[p].l].l].size>t[t[p].r].size)//LL
rRotate(p);
else if(t[t[t[p].l].r].size>t[t[p].r].size)//LR
lRotate(t[p].l),
rRotate(p);
else return ;
}
else
{
if(t[t[t[p].r].r].size>t[t[p].l].size)
lRotate(p);
else if(t[t[t[p].r].l].size>t[t[p].l].size)
rRotate(t[p].r),
lRotate(p);
else return ;
}
maintain(t[p].l, false);
maintain(t[p].r, true);
maintain(p, false);
maintain(p, true);
}
void insert(int &p,int key,int id)
{
if(p==0)
{
p=++top;
t[p].l=t[p].r=0;
t[p].size=1;
t[p].key=key,t[p].id=id;
}
else
{
t[p].size++;
if(key<t[p].key) insert(t[p].l,key,id);
else insert(t[p].r,key,id);
maintain(p, key>=t[p].key);
}
}
int remove(int &p,int key)
{
t[p].size--;
if(t[p].key==key||
(key<t[p].key&&t[p].l==0)||
(key>t[p].key&&t[p].r==0))
{
int retkey=t[p].key;
if(t[p].l&&t[p].r)
t[p].key=remove(t[p].l,t[p].key+1);
else
p=t[p].l+t[p].r;
return retkey;
}
else if(key<t[p].key) return remove(t[p].l,key);
else return remove(t[p].r,key);
}
int getmin()
{
int p=root;
while(t[p].l) p=t[p].l;
printf("%d\n",t[p].id);
return t[p].key;
}
int getmax()
{
int p=root;
while(t[p].r) p=t[p].r;
printf("%d\n",t[p].id);
return t[p].key;
}
int main()
{
top=root=0;
int op,id,key;
while(scanf("%d",&op)!=EOF&&op)
{
if(op==1)
{
scanf("%d%d",&id,&key);
insert(root,key,id);
}
else if(op==2)
{
int temp=getmax();
remove(root,temp);
}
else
{
int temp=getmin();
remove(root,temp);
}
}
return 0;
}