[luogu 3369]普通平衡树（fhq

题目描述

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

插入x数
删除x数(若有多个相同的数，因只删除一个)
查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数，因输出最小的排名)
查询排名为x的数
求x的前驱(前驱定义为小于x，且最大的数)
求x的后继(后继定义为大于x，且最小的数)

输入输出格式

输入格式：

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号( 1≤opt≤6 1 \leq opt \leq 6 1≤opt≤6 )

输出格式：

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

输入输出样例

输入样例#1： 复制

输出样例#1： 复制

说明

时空限制：1000ms,128M

1.n的数据范围： n≤100000 n \leq 100000 n≤100000

2.每个数的数据范围： $[-{10}^7, {10}^7]$

题解：

模板题，用的是fhq_treap。

 //Never forget why you start

 #include<iostream>

 #include<cstdio>

 #include<cstring>

 #include<cstdlib>

 #include<cmath>

 #include<algorithm>

 #include<ctime>

 #define ll(x) tre[x].child[0]

 #define rr(x) tre[x].child[1]

 #define son(x,t) tre[x].child[t]

 using namespace std;

 int n,m,cnt,root;

 struct Treap{

   int child[],x,size,rev;

 }tre[];

 void push_up(int root){

   tre[root].size=tre[ll(root)].size+tre[rr(root)].size+;

 }

 int newnode(int x){

   cnt++;

   tre[cnt].size=;

   tre[cnt].rev=rand();

   tre[cnt].x=x;

   return cnt;

 }

 void split(int now,int k,int &x,int &y){

   if(!now)x=y=;

   else{

     if(tre[now].x<=k)

       x=now,split(rr(now),k,rr(now),y);

     else

       y=now,split(ll(now),k,x,ll(now));

     push_up(now);

   }

 }

 int merge(int x,int y){

   if(!x||!y)return x+y;

   if(tre[x].rev<tre[y].rev){

     rr(x)=merge(rr(x),y);

     push_up(x);

     return x;

   }

   else{

     ll(y)=merge(x,ll(y));

     push_up(y);

     return y;

   }

 }

 int find(int root,int k){

   int y=tre[ll(root)].size;

   if(y+==k)return root;

   else if(y>=k)return find(ll(root),k);

   else find(rr(root),k-y-);

 }

 int main(){

   int i,j,a,b,c;

   srand(time());

   scanf("%d",&n);

   while(n--){

     scanf("%d%d",&i,&j);

     if(i==){

       split(root,j,a,b);

       root=merge(merge(a,newnode(j)),b);

     }

     if(i==){

       split(root,j,a,c);

       split(a,j-,a,b);

       b=merge(ll(b),rr(b));

       root=merge(merge(a,b),c);

     }

     if(i==){

       split(root,j-,a,b);

       printf("%d\n",tre[a].size+);

       root=merge(a,b);

     }

     if(i==){

       printf("%d\n",tre[find(root,j)].x);

     }

     if(i==){

       split(root,j-,a,b);

       printf("%d\n",tre[find(a,tre[a].size)].x);

       root=merge(a,b);

     }

     if(i==){

       split(root,j,a,b);

       printf("%d\n",tre[find(b,)].x);

       root=merge(a,b);

     }

   }

   return ;

 }

巴特西

[luogu 3369]普通平衡树（fhq_treap）

题目描述

输入输出格式

输入输出样例

说明

题解：

最新文章

热门文章