首页 > 代码库 > poj 3481 平衡树
poj 3481 平衡树
裸的treap
#include<stdio.h> #include<algorithm> #include<stdlib.h> #include<cstring> #include<iostream> #include<string> #include<cmath> #include<vector> #include<queue> #include<map> #include<iterator> #include<stack> using namespace std; #define ll __int64 #define MAXN 500010 #define inf 2000000007 struct node { struct node *ch[2]; int sz,key,num,pri; }; typedef node * tree; node base[MAXN],nil; tree root,null,top; void Init() { top=base; root=null=&nil; null->ch[0]=null->ch[1]=null; null->key=null->pri=null->num=2147483647; null->sz=0; } inline int random() { static int seed=703; return seed=int(seed *48271ll %2147483647); } inline tree newnode(int num,int key) { top->num=num; top->key=key; top->pri=random(); top->sz=1; top->ch[0]=top->ch[1]=null; return top++; } void Rotate(tree &x,bool d) { tree y = x->ch[!d]; x->ch[!d]=y->ch[d]; y->ch[d]=x; x->sz=x->ch[0]->sz+1+x->ch[1]->sz; y->sz=y->ch[0]->sz+1+y->ch[1]->sz; x=y; } void Insert(tree &t,int num,int key) { if(t==null) t=newnode(num,key); else { bool d=key>t->key; Insert(t->ch[d],num,key); (t->sz)++; if(t->pri<t->ch[d]->pri) Rotate(t,!d); } } void Delete(tree &t,int key) { if(t->key!=key) { Delete(t->ch[key>t->key],key); (t->sz)--; } else if(t->ch[0]==null) t=t->ch[1]; else if(t->ch[1]==null) t=t->ch[0]; else { bool d=t->ch[0]->pri<t->ch[1]->pri; Rotate(t,d); Delete(t->ch[d],key); } } tree select(int k) //第k个 { tree t=root; int tmp; for(;;) { tmp=t->ch[0]->sz+1; if(k==tmp) return t; else if(k>tmp) { k-=tmp; t=t->ch[1]; } else t=t->ch[0]; } } int cnt; int main() { Init(); int ty; cnt=0; while(scanf("%d",&ty)!=EOF) { if(ty==0) break; if(ty==1) { int num,key; scanf("%d%d",&num,&key); Insert(root,num,key); cnt++; } else if(ty==2) { if(cnt>0) { tree a=select(cnt); int b=a->key; //printf("%d\n",b); Delete(root,b); printf("%d\n",a->num); cnt--; } else printf("0\n"); } else { if(cnt>0) { tree a=select(1); int b=a->key; Delete(root,b); printf("%d\n",a->num); cnt--; } else printf("0\n"); } } return 0; }
poj 3481 平衡树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。