3224: Tyvj 1728 普通平衡树
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 15689 Solved: 6832
[Submit][Status][Discuss]
Description
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
1. 插入x数
2. 删除x数(若有多个相同的数,因只删除一个)
3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x,且最大的数)
6. 求x的后继(后继定义为大于x,且最小的数)
Input
第一行为n,表示操作的个数,下面n行每行有两个数opt和x,opt表示操作的序号(1<=opt<=6)
Output
对于操作3,4,5,6每行输出一个数,表示对应答案
Sample Input
10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
Sample Output
106465
84185
492737
84185
492737
HINT
1.n的数据范围:n<=100000
2.每个数的数据范围:[-2e9,2e9]
Source
为了做紫荆花之恋学习了scapegoat tree
代码要比treap splay 复杂一点 但很好理解
就是每次insert时一看不平衡就暴力重建 (是不是很暴力啊)
这个东西真是很快啊 Z基和lkh的treap和splay被我虐爆了 哈哈哈
#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef double db;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}
const int N=2001000,inf=0X3f3f3f3f;
const db alpha=0.75;
struct scapegoat_tree{int s[2],fa,sz,val;}tr[N];
int root,tot;
void initial()
{
tot=2;root=1;
tr[1].val=-inf;tr[1].sz=2;tr[1].s[1]=2;
tr[2].val=inf;tr[2].sz=1;tr[2].fa=1;
}
bool balance(int k)
{return (alpha*tr[k].sz>=(db)tr[tr[k].s[0]].sz)&&(alpha*tr[k].sz>=(db)tr[tr[k].s[1]].sz);}
int sta[N],top;
void seek(int k)
{if(!k)return;seek(tr[k].s[0]);sta[++top]=k;seek(tr[k].s[1]);}
int build(int l,int r)
{
if(l>r)return 0;
int mid=(l+r)>>1,now=sta[mid];
tr[now].s[0]=build(l,mid-1);
tr[now].s[1]=build(mid+1,r);
tr[tr[now].s[1]].fa=tr[tr[now].s[0]].fa=now;
tr[now].sz=tr[tr[now].s[0]].sz+tr[tr[now].s[1]].sz+1;
return now;
}
void rebuild(int k)
{
top=0;seek(k);
int fa=tr[k].fa,son=(tr[fa].s[1]==k),now=build(1,top);
tr[now].fa=fa;
tr[fa].s[son]=now;
if(root==k)root=now;
}
void insert(int x)
{
register int now=root,inv=0,i;
register bool son;
tr[++tot].val=x;tr[tot].sz=1;
while(1)
{
tr[now].sz++;
son=(x>=tr[now].val);
if(tr[now].s[son])now=tr[now].s[son];
else {tr[tot].fa=now;tr[now].s[son]=tot;break;}
}
for(i=tot;i;i=tr[i].fa)if(!balance(i))inv=i;
if(inv)rebuild(inv);
}
inline int get_pos(int x)
{
int now=root;
while(tr[now].val^x)
x<tr[now].val?now=tr[now].s[0]:now=tr[now].s[1];
return now;
}
void del(int k)
{
if(tr[k].s[0]&&tr[k].s[1])
{
int aim=tr[k].s[0];
while(tr[aim].s[1])aim=tr[aim].s[1];
tr[k].val=tr[aim].val;
k=aim;
}
int tmp=tr[k].s[0]?tr[k].s[0]:tr[k].s[1],son=(tr[tr[k].fa].s[1]==k);
tr[tr[k].fa].s[son]=tmp;tr[tmp].fa=tr[k].fa;
if(root==k)root=tmp;
while(tr[k].fa)k=tr[k].fa,tr[k].sz--;
}
inline int query_rank(int x)
{
int now=root,res=0;
while(now)
tr[now].val<x?(res+=tr[tr[now].s[0]].sz+1,now=tr[now].s[1]):now=tr[now].s[0];
return res;
}
inline int query_kth(int kth)
{
int now=root;
while(1)
{
if(tr[tr[now].s[0]].sz+1==kth)return now;
tr[tr[now].s[0]].sz>=kth?now=tr[now].s[0]:(kth-=tr[tr[now].s[0]].sz+1,now=tr[now].s[1]);
}
}
inline int query_pre(int x)
{
int now=root,res;
while(now)
tr[now].val<x?(res=tr[now].val,now=tr[now].s[1]):now=tr[now].s[0];
return res;
}
inline int query_sub(int x)
{
int now=root,res;
while(now)
tr[now].val>x?(res=tr[now].val,now=tr[now].s[0]):now=tr[now].s[1];
return res;
}
int main()
{
register int n,opt,x;
initial();
n=read();
while(n--)
{
opt=read();x=read();
switch(opt)
{
case 1:insert(x);break;
case 2:del(get_pos(x));break;
case 3:print(query_rank(x));puts("");break;
case 4:print(tr[query_kth(x+1)].val);puts("");break;
case 5:print(query_pre(x));puts("");break;
case 6:print(query_sub(x));puts("");break;
}
}
return 0;
}
/*
8
1 10
1 20
1 30
3 20
4 2
2 10
5 25
6 -1
2
20
20
20
*/