bzoj 3600 没有人的算术——二叉查找树动态标号

时间:2023-03-10 01:56:57
bzoj 3600 没有人的算术——二叉查找树动态标号

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3600

已知 l 和 r 的排名,想快速知道 k 的排名。那么建一个 BIT ,用已知的排名做比较,走到一个位置,就知道自己的排名了。

但这样会让很多点的排名改变。如果用实数表示标号就可以方便地生成一个 “两个排名之间的排名” 而不影响其他点。

BIT 的每个节点代表一个曾经出现过的序列里的点,有自己的实数标号和 ( l , r ) 的组成。

如果把实数标号不是记在 BIT 节点上而是记在序列上,即 v[ i ] 表示序列第 i 个位置的标号,那么修改 k 位置的话,其他曾经调用 k 而组成的点也会收到影响。

所以实数标号记在 BIT 节点上,序列记一个 dy[ ] 表示它对应 BIT 哪个点。序列一个位置的 ( l , r ) 记录成 BIT 上对应的点即可。

用替罪羊树实现平衡。那么在重构的时候要注意,因为 dy[ ] 不会改变,所以 BIT 上点标之间的大小关系也不应改变。只要在回收的时候不是回收 Node 而是回收 int 即可(int 是那个 Node 的角标)。

最开始的时候只有一个 0 。那么 BIT 里唯一的节点表示 “只有一个 0 ” ,其 l 就是自己, r 是 0 (BIT 节点从1标号)。v[ 1 ] 会是 0.5 (如果标号总范围是 ( 0 , 1 ) 的话), v[ 0 ] 会是 0 ,这样就表示了 “没有” 是最小的。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define db double
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
int rdn()
{
int ret=;bool fx=;char ch=getchar();
while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
return fx?ret:-ret;
}
const int N=1e5+,M=5e5+;
const db sl=0.75;
int n,rt,tot,c0[M],c1[M],siz[M],dy[N]; db v[M];
int Ls[N<<],Rs[N<<],mx[N<<];
int rcr,rfa,sta[M],top; bool rfx; db rL,rR;
struct Node{
int l,r;
Node(int l=,int r=):l(l),r(r) {}
bool operator< (const Node &b)const
{ return v[l]==v[b.l]?v[r]<v[b.r]:v[l]<v[b.l];}
bool operator== (const Node &b)const
{ return v[l]==v[b.l]&&v[r]==v[b.r];}
}a[M];
int Mx(int u,int v){return u>v?u:v;}
Node Mx(Node u,Node v){return u<v?v:u;}
void build(int l,int r,int cr)
{
mx[cr]=l; if(l==r)return;
int mid=l+r>>;
ls=++tot; build(l,mid,ls);
rs=++tot; build(mid+,r,rs);
}
void mdfy(int l,int r,int cr,int p)
{
if(l==r)return; int mid=l+r>>;
if(p<=mid)mdfy(l,mid,ls,p);
else mdfy(mid+,r,rs,p);
if(a[dy[mx[ls]]]<a[dy[mx[rs]]])mx[cr]=mx[rs];
else mx[cr]=mx[ls];
}
int qry(int l,int r,int cr,int L,int R)
{
if(l>=L&&r<=R)return mx[cr];
int mid=l+r>>;
if(R<=mid)return qry(l,mid,ls,L,R);
if(mid<L)return qry(mid+,r,rs,L,R);
int u=qry(l,mid,ls,L,R), v=qry(mid+,r,rs,L,R);
if(a[dy[u]]<a[dy[v]])return v;
else return u;
}
void chk(int cr,db L,db Md,db R)
{
if(Mx(siz[c0[cr]],siz[c1[cr]])>=siz[cr]*sl)rcr=cr;
if(rcr==c0[cr])
{ rfa=cr; rfx=; rL=L; rR=Md;}
if(rcr==c1[cr])
{ rfa=cr; rfx=; rL=Md; rR=R;}
}
int ins(int &cr,Node k,db L,db R)
{
db Md=(L+R)/;
if(!cr)
{
cr=++tot; a[cr]=k;
v[cr]=Md;siz[cr]=;return cr;
}
if(k==a[cr])return cr;
int ret=; siz[cr]++;
if(k<a[cr])ret=ins(c0[cr],k,L,Md);
else ret=ins(c1[cr],k,Md,R);
chk(cr,L,Md,R); return ret;
}
void del_dfs(int cr)//sorted!!
{
if(c0[cr])del_dfs(c0[cr]);
sta[++top]=cr;
if(c1[cr])del_dfs(c1[cr]);
}
void bld(int l,int r,int &cr,db L,db R)
{
db Md=(L+R)/; int mid=l+r>>;
cr=sta[mid];
c0[cr]=c1[cr]=; siz[cr]=; v[cr]=Md;
if(l<mid)bld(l,mid-,c0[cr],L,Md);
if(mid<r)bld(mid+,r,c1[cr],Md,R);
siz[cr]=siz[c0[cr]]+siz[c1[cr]]+;
}
void rebld(int &cr,db L,db R)
{
top=; del_dfs(cr); bld(,top,cr,L,R);
}
int main()
{
n=rdn();int m=rdn(); char ch[];
tot=;build(,n,);
tot=; ins(rt,Node(,),,);//v[1]=0.5//v[0]=0 is min
for(int i=;i<=n;i++)dy[i]=;
for(int i=,l,r,k;i<=m;i++)
{
scanf("%s",ch); l=rdn();r=rdn();
if(ch[]=='C')
{
k=rdn();
dy[k]=ins(rt,Node(dy[l],dy[r]),,);
Node tp=a[dy[k]];
mdfy(,n,,k);
if(rcr)
{
if(rcr==rt)rebld(rt,,);
else if(!rfx)rebld(c0[rfa],rL,rR);
else rebld(c1[rfa],rL,rR);
rcr=;
}
}
else printf("%d\n",qry(,n,,l,r));
}
return ;
}