2021.12.08 [SHOI2009]会场预约(平衡树游码表)

2023-05-13,,

2021.12.08 [SHOI2009]会场预约(平衡树游码表

https://www.luogu.com.cn/problem/P2161

题意:

你需要维护一个 在数轴上的线段 的集合 \(S\),支持两种操作:

A l r 表示将 \(S\) 中所有与线段 \([l,r]\) 相交的线段删去,并将 \([l,r]\) 加入 \(S\) 中。
B 查询 \(S\) 中的元素数量。

对于 A 操作,每次还需输出删掉的元素个数。

分析:

对于这道题,我先想到了珂朵莉树(毕竟这个名字太深入人心,虽然实际就是游码表),然后发现……坏了,忘了怎么打了,于是转战平衡树。

把FHQ Treap存的关键码变成俩,一个存 \(l\) ,一个存 \(r\) 。

对于操作A:首先按照 \(l\) ,以 \(l+1\) 为标准,分离出 \([1,l]\) 与 \([l+1,n]\) ,然后再按照 \(r\) ,以 \(r\) 为标准,分离出 \([1,r-1]\) 与 \([r,n]\) ,对于中间部分输出大小,加入新点,合并。

对于操作B:输出平衡树总大小。

代码如下:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<set>
#define IOS ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std; const int N=2e5+10;
int n,root,cnt,sizei[N],son[N][2],val[N][2],key[N]; inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){
if(ch=='-')w=-1;
ch=getchar();
}
while(ch<='9'&&ch>='0'){
s=s*10+ch-'0';
ch=getchar();
}
return s*w;
}
inline int add(int x,int y){
++cnt;
sizei[cnt]=1;
val[cnt][0]=x;val[cnt][1]=y;
key[cnt]=rand();
return cnt;
}
inline void update(int x){
sizei[x]=sizei[son[x][0]]+sizei[son[x][1]]+1;
}
inline void split(int rt,int k,int flag,int &x,int &y){
if(!rt)return (void)(x=y=0);
if(val[rt][flag]<k)
x=rt,split(son[rt][1],k,flag,son[rt][1],y);
else y=rt,split(son[rt][0],k,flag,x,son[rt][0]);
update(rt);
}
inline int merge(int x,int y){
if(!x||!y)return x+y;
if(key[x]>=key[y]){
son[x][1]=merge(son[x][1],y);
update(x);
return x;
}else{
son[y][0]=merge(x,son[y][0]);
update(y);
return y;
}
} int main(){
n=read();
for(int i=1;i<=n;i++){
char ch;
cin>>ch;
if(ch=='A'){
int u,v;
u=read();v=read();
int x,y,z;
split(root,v+1,0,x,z);
split(x,u,1,x,y);
cout<<sizei[y]<<endl;
root=merge(merge(x,add(u,v)),z);
}else cout<<sizei[root]<<endl;
}
return 0;
}

2021.12.08 [SHOI2009]会场预约(平衡树游码表)的相关教程结束。

《2021.12.08 [SHOI2009]会场预约(平衡树游码表).doc》

下载本文的Word格式文档,以方便收藏与打印。