treap模板(tyvj普通二叉树)

模板题。。。
#include <cassert>
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cctype>
#include <cmath>
#include <ctime>
using namespace std;
#define INF 0x3fffffff
struct Treap
{
struct node{int v,tot,sz,rnd,ch[2];}t;
int sz,root,tmp;
void init(long long int n){t=(node
)malloc(sizeof(node)*n+10);}
void update(int x){t[x].sz=t[t[x].ch[0]].sz+t[t[x].ch[1]].sz+t[x].tot;}
void turn(int &x,int k)
{
int y=t[x].ch[k^1];
t[x].ch[k^1]=t[y].ch[k];
t[y].ch[k]=x;
update(x);
update(y);
x=y;
}
void insert(int &x,int v)
{
if(!x)
t[++sz].v=v,t[sz].rnd=rand(),t[sz].tot=t[sz].sz=1,x=sz;
else{
if(t[x].sz++,t[x].v==v)t[x].tot++;
else if(insert(t[x].ch[tmp=v>t[x].v],v),t[t[x].ch[tmp]].rnd>t[x].rnd)turn(x,tmp^1);
}
}
void del(int &x,int v)
{
if(!x)return;
if(t[x].v==v){
if(t[x].tot>1)t[x].tot--,t[x].sz--;
else{
if(!(t[x].ch[0]&&t[x].ch[1]))x=t[x].ch[0]|t[x].ch[1];
else turn(x,tmp=t[t[x].ch[0]].rnd>t[t[x].ch[1]].rnd),t[x].sz--,del(t[x].ch[tmp],v);
}
}
else
t[x].sz--,del(t[x].ch[v>t[x].v],v);
}
int askRank(int x,int v)
{
if(t[x].v==v)return t[t[x].ch[0]].sz+1;
if(t[x].v<v)return t[t[x].ch[0]].sz+t[x].tot+askRank(t[x].ch[1],v);
else return askRank(t[x].ch[0],v);
}
int askKth(int x,int v)
{
if(!x)return 0;
if(v<=t[t[x].ch[0]].sz)return askKth(t[x].ch[0],v);
else if(v>t[t[x].ch[0]].sz+t[x].tot)return askKth(t[x].ch[1],v-t[t[x].ch[0]].sz-t[x].tot);
else return t[x].v;
}
int pre(int v)
{
insert(root,v);
tmp=askKth(root,askRank(root,v)-1);
del(root,v);
return tmp;
}
int find(int x,int v){return t[x].v==v?x:find(t[x].ch[t[x].v<v],v);}
int sub(int v)
{
insert(root,v);
tmp=askKth(root,askRank(root,v)+t[find(root,v)].tot);
del(root,v);
return tmp;
}
}treap;
int main()
{
int n;
scanf("%d",&n);
treap.init(n);
while(n--)
{
int a,b;
scanf("%d%d",&a,&b);
switch(a)
{
case 1:treap.insert(treap.root,b);break;//1. 插入x数
case 2:treap.del(treap.root,b);break;//2. 删除x数(若有多个相同的数,因只删除一个)
case 3:printf("%d\n",treap.askRank(treap.root,b));break;//3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
case 4:printf("%d\n",treap.askKth(treap.root,b));break;//4. 查询排名为x的数
case 5:printf("%d\n",treap.pre(b));break;//5. 求x的前驱(前驱定义为小于x,且最大的数)
case 6:printf("%d\n",treap.sub(b));break;//6. 求x的后继(后继定义为大于x,且最小的数)
}
}
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,390评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,821评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,632评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,170评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,033评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,098评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,511评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,204评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,479评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,572评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,341评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,213评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,576评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,893评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,171评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,486评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,676评论 2 335

推荐阅读更多精彩内容