首页 > 其他分享 >BZOJ 3224 普通平衡树 (BST+Treap)

BZOJ 3224 普通平衡树 (BST+Treap)

时间:2023-02-03 11:00:11浏览次数:46  
标签:return val Treap BST 3224 Update MMM MM int


题目描述:

您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:

  1. 插入数值x。
  2. 删除数值x(若有多个相同的数,应只删除一个)。
  3. 查询数值x的排名(若有多个相同的数,应输出最小的排名)。
  4. 查询排名为x的数值。
  5. 求数值x的前驱(前驱定义为小于x的最大的数)。
  6. 求数值x的后继(后继定义为大于x的最小的数)。

输入格式

第一行为n,表示操作的个数。

接下来n行每行有两个数opt和x,opt表示操作的序号(1<=opt<=6)。

输出格式

对于操作3,4,5,6每行输出一个数,表示对应答案。

数据范围

n≤100000n≤100000,所有数均在−107−107到107107内。

输入样例:

8 1 10 1 20 1 30 3 20 4 2 2 10 5 25 6 -1

输出样例:

2 20 20 20

这道题是最基本的BST+Treap板子题,直接套板子然后考虑细节就行,

关于算法可以参考这两篇博客

​​BST​​

​​Treap​​

AC代码:

/*

MMM
MMMMMMM:
~MM MMM
7M+ MM
MM
MMMMMM, MM MM ,,:+MMM:
MMM MMM MM~ MM ~MMMMMMMMMMMMMMMMMMMMMMM$
ZM$ MMM MM MMMMMMMMMN MMM MMM$
MM MM8 MM OMMMMMO MM MM~
,MM MM ,MMMMM MM MMMMMMZ ,MM MMM
MM? MMDMMMMM+ , MM MM, ~MD , MM
MM8 :MMMO IMMM, MMÁÁÁÁ MM MM MMM ,MMO MM
MMM MMMMM MMMMMMM MM M8 MM MMM MM NM
MMMMMM MM ~MM MM7 MM MM7 DM
7MMM MM ÁÁÁÁ MM MMMMMM MM? MM
MMM M MM MM MMD :MM
MMM MMM IMM NMMM IMM
NMM 7MMMMMM MMMMMMMMM
MMM
MM MMM
MM DMM? MMM:
7MM MMMMMMMMMM MMMM
MM MMM MMMN MMMMMM7
MM MM MMD MMMMM
DM~ MM MM MM
NM MM MM MM=
MM MM NM MM
NM +MM MM NM MM
OM? MMM 8MM ,MM ZMZ
MM MMMM :MMM +? MMM MM
MM MMMMMMM: MM =MM +MM
MM: MMM MMM MM
MM MMMMMMMMMMMMM? MM
MM+ ?MMMMM$ MM7
MM MM
MM MM
=MM ZMM
MMM MMMM
~MMMMM MMMMMM
MM ?MMMM 8MMMM MM
MM MMMMMMN, NMMMMM ,MM
ZMM =MMMMMMMMMMMMMM DMM
,MMMMOMM8 MM7MMMM
MMM? MMM, MMMM ,MMM +MMM MMM
MM MMMM 7MMM MM MM ?MMM 8MMM8 M
:MMM NMM~ MM MM MMM MMMM
MO MM MM MM DMI ?ND
~M,ZMNIM MM 8MM :M M$ MM
MM MM MM MOM8MMM
~ MM MM D
OM? ~MM
MMMM MM MM
M MM M= MM =MM
MM M~MMMMNMMN MN
M M,MM+M?MM MM
MNOMMM ,,NDMM MM
MMM MM
MM MM
+M$ MM
MM MM
MM MM
MMMMMMMMMMMMMMMMMMMMMMMMMMNNNNNNDDZZZZIIIIIIIZZZDNMMMMM
, :~IZ8DNMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMMN
NM MZ MM M
NM M7 MM7M
NM M+ MMDM
DM M= MMMM
DM M MMMM ZM=
DMMMMMMMMMMMMN MMMMMMMMMMMMM=
MM MM MM MM
~MMMMMMMMMMMM8 +MMMMMMMMMMMM
*/

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stdlib.h>
#include<queue>
#include<map>
#include<set>
#include<iomanip>
#include<math.h>
using namespace std;
typedef long long ll;
typedef double ld;
const int INF = 0x3f3f3f3f;
const int N = 100010;

struct Treap
{
int l, r;
int val, dat;
int cnt, size;
} a[N];

int tot, root, n;

int New(int val)
{
a[++tot].val=val;
a[tot].dat=rand();
a[tot].cnt=a[tot].size=1;
return tot;
}

void Update(int p)
{
a[p].size=a[a[p].l].size+a[a[p].r].size+a[p].cnt;
}

void Build()
{
New(-INF),New(INF);
root=1,a[1].r=2;
Update(root);
}

int GetRankByVal(int p, int val)
{
if (p==0)
return 0;
if (val==a[p].val)
return a[a[p].l].size + 1;
if (val<a[p].val)
return GetRankByVal(a[p].l,val);
return GetRankByVal(a[p].r,val)+a[a[p].l].size+a[p].cnt;
}

int GetValByRank(int p, int rank)
{
if(p==0)
return INF;
if(a[a[p].l].size>=rank)
return GetValByRank(a[p].l,rank);
if (a[a[p].l].size + a[p].cnt>=rank)
return a[p].val;
return GetValByRank(a[p].r,rank-a[a[p].l].size-a[p].cnt);
}

void zig(int &p)
{
int q=a[p].l;
a[p].l=a[q].r;
a[q].r=p;
p=q;
Update(a[p].r);
Update(p);
}

void zag(int &p)
{
int q=a[p].r;
a[p].r=a[q].l;
a[q].l=p;
p=q;
Update(a[p].l);
Update(p);
}

void Insert(int &p, int val)
{
if(p==0)
{
p=New(val);
return ;
}
if(val==a[p].val)
{
a[p].cnt++;
Update(p);
return ;
}
if(val<a[p].val)
{
Insert(a[p].l,val);
if (a[p].dat<a[a[p].l].dat)
zig(p);
}
else
{
Insert(a[p].r,val);
if (a[p].dat<a[a[p].r].dat)
zag(p);
}
Update(p);
}

int Getpre(int val)
{
int ans=1;
int p=root;
while(p)
{
if(val==a[p].val)
{
if(a[p].l>0)
{
p=a[p].l;
while(a[p].r>0)
p=a[p].r;
ans=p;
}
break;
}
if(a[p].val<val&&a[p].val>a[ans].val)
ans=p;
p=val<a[p].val?a[p].l:a[p].r;
}
return a[ans].val;
}

int Getnext(int val)
{
int ans=2;
int p=root;
while(p)
{
if(val==a[p].val)
{
if(a[p].r>0)
{
p=a[p].r;
while(a[p].l>0)
p=a[p].l;
ans=p;
}
break;
}
if(a[p].val>val&&a[p].val<a[ans].val)
ans=p;
p=val<a[p].val?a[p].l:a[p].r;
}
return a[ans].val;
}

void Remove(int &p, int val)
{
if(p==0)
return ;
if(val==a[p].val)
{
if(a[p].cnt>1)
{
a[p].cnt--;
Update(p);
return;
}
if (a[p].l||a[p].r)
{
if(a[p].r==0||a[a[p].l].dat>a[a[p].r].dat)
{
zig(p);
Remove(a[p].r,val);
}
else
{
zag(p);
Remove(a[p].l,val);
}
Update(p);
}
else p = 0;
return ;
}
val<a[p].val?Remove(a[p].l,val):Remove(a[p].r,val);
Update(p);
}

int main()
{
Build();
cin >> n;
while(n--)
{
int opt,x;
cin>>opt>>x;
switch (opt)
{
case 1:
Insert(root,x);
break;
case 2:
Remove(root,x);
break;
case 3:
cout<<GetRankByVal(root,x)-1<<endl;
break;
case 4:
cout<<GetValByRank(root,x+1)<<endl;
break;
case 5:
cout<<Getpre(x)<<endl;
break;
case 6:
cout<<Getnext(x)<<endl;
break;
}
}
return 0;
}

 

标签:return,val,Treap,BST,3224,Update,MMM,MM,int
From: https://blog.51cto.com/u_15952369/6035597

相关文章

  • Treap 平衡二叉查找树
    【基本概念】Treap=Tree+Heap。Tree是指二叉搜索树,而Heap指的是二叉堆,一般是最小堆。Treap需要维护两值,一个是二叉搜索树中的键值(key),另一个是最小堆中的优先级(aux)。Treap是......
  • BST 二叉搜索树
    BST,又叫平衡二叉树,是一种循关键码访问的二叉树,每个节点带有一个数值就是关键码,并且要求保持顺序性,即任一节点不小于其左后代,不大于其右后代(注意是后代,不是孩子)。BST的顺序性......
  • POJ2761 Feed the dogs(Treap)
    DescriptionWindlovesprettydogsverymuch,andshehasnpetdogs.SoJiajiahastofeedthedogseverydayforWind.JiajialovesWind,butnotthedogs,......
  • BZOJ1503 郁闷的出纳员 (Treap)
    DescriptionOIER公司是一家大型专业化软件公司,有着数以万计的员工。作为一名出纳员,我的任务之一便是统计每位员工的工资。这本来是一份不错的工作,但是令人郁闷的是,我们......
  • 使用webStrom使用uni-app新建微信小程序
    一、通过官网了解uni-app相关的信息uni-app官网地址:https://uniapp.dcloud.net.cn/已知uni-app微信小程序程序创建需要使用HBuilderX或者cli进行创建,再到微信开发者工具......
  • 【计算机网络】Stanford CS144 Lab1 : stitching substrings into a byte stream 学
    Puttingsubstringsinsequence实现一个流重组器。可以将带有索引的流碎片按照顺序重组。这些流碎片是可以重复的部分,但是不会有冲突的部分。这些流碎片将通过Lab0中......
  • WebStorm注册码2023年安装教程
    WebStorm是一款非常流行的JavaScript集成开发环境(IDE),用于构建Web和Node.js应用程序。它提供了丰富的功能,如代码编辑、调试、代码检查和重构工具,可以帮助开发人员提高生产力......
  • abstract抽象类
    1.//abstract抽象方法,只有方法名字,没有方法的实现!publicabstractvoiddoSomething();2.抽象类的所有方法,继承了他的子类,都必须要实现他的方法~除非他的子类也是抽象......
  • 解决webstorm不能识别vue的@路径引用
    方式1: webstorm无法识别@符号问题[ctrl+左键无法跳转]解决:创建vue项目,打开项目后,再进入webstorm配置页面[否则没有webpack选项]ctrl+alt+s[进入设置页面]->langua......
  • Abstraction, intuition, and the “monad tutorial fallacy”
    WhileworkingonanarticlefortheMonad.Reader,I’vehadtheopportunitytothinkabouthowpeoplelearnandgainintuitionforabstraction,andtheimplica......