[BZOJ3123][Sdoi2013]森林（主席树启发式合并）

题目描述

题解

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 100005
#define lg 17

int T,n,m,q,x,y,k,s,t,sz,ans;
int tot,point
,nxt[N*2],v[N*2];
int a
,b
;
int ufs
,size
,h
,f
[lg+5];
int root
,sum[N*200],ls[N*200],rs[N*200];
bool vis
;

int lsh(int x)
{
int l=1,r=b[0],mid,ans;
while (l<=r)
{
mid=(l+r)>>1;
if (b[mid]>=x) ans=mid,r=mid-1;
else l=mid+1;
}
return ans;
}
int find(int x)
{
if (ufs[x]==x) return x;
ufs[x]=find(ufs[x]);
return ufs[x];
}
void add(int x,int y)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void update(int &now,int l,int r,int x)
{
int mid=(l+r)>>1;
sum[++sz]=sum[now]+1,ls[sz]=ls[now],rs[sz]=rs[now];now=sz;
if (l==r) return;
if (x<=mid) update(ls[now],l,mid,x);
else update(rs[now],mid+1,r,x);
}
int query(int x,int y,int s,int t,int l,int r,int k)
{
int mid=(l+r)>>1;
if (l==r) return l;
int tmp=sum[ls[x]]+sum[ls[y]]-sum[ls[s]]-sum[ls[t]];
if (tmp>=k) return query(ls[x],ls[y],ls[s],ls[t],l,mid,k);
else return query(rs[x],rs[y],rs[s],rs[t],mid+1,r,k-tmp);
}
void build(int x,int fa)
{
vis[x]=true;
h[x]=h[fa]+1;
root[x]=root[fa];
update(root[x],1,b[0],a[x]);
for (int i=1;i<lg;++i)
{
if (h[x]-(1<<i)<1) break;
f[x][i]=f[f[x][i-1]][i-1];
}
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa)
{
int f1=find(x),f2=find(v[i]);
ufs[f2]=f1;size[f1]+=size[f2];
f[v[i]][0]=x;
build(v[i],x);
}
}
int lca(int x,int y)
{
if (h[x]<h[y]) swap(x,y);
int k=h[x]-h[y];
for (int i=0;i<lg;++i)
if ((k>>i)&1) x=f[x][i];
if (x==y) return x;
for (int i=lg-1;i>=0;--i)
if (f[x][i]!=f[y][i])
x=f[x][i],y=f[y][i];
return f[x][0];
}
void rebuild(int x,int fa)
{
h[x]=h[fa]+1;
root[x]=root[fa];
update(root[x],1,b[0],a[x]);
for (int i=1;i<lg;++i)
{
//      if (h[x]-(1<<i)<1) break;
f[x][i]=f[f[x][i-1]][i-1];
}
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa)
{
f[v[i]][0]=x;
rebuild(v[i],x);
}
}
int main()
{
scanf("%d",&T);
scanf("%d%d%d",&n,&m,&q);
for (int i=1;i<=n;++i) scanf("%d",&a[i]),b[i]=a[i];
sort(b+1,b+n+1);
b[0]=unique(b+1,b+n+1)-b-1;
for (int i=1;i<=n;++i) a[i]=lsh(a[i]);
for (int i=1;i<=m;++i)
{
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
}
for (int i=1;i<=n;++i) ufs[i]=i,size[i]=1;
for (int i=1;i<=n;++i)
if (!vis[i])
build(i,0);
for (int i=1;i<=q;++i)
{
char opt=getchar();
while (opt!='L'&&opt!='Q') opt=getchar();
if (opt=='Q')
{
scanf("%d%d%d",&x,&y,&k);
x^=ans,y^=ans,k^=ans;
s=lca(x,y);t=f[s][0];
ans=query(root[x],root[y],root[s],root[t],1,b[0],k);
ans=b[ans];
printf("%d\n",ans);
}
else
{
scanf("%d%d",&x,&y);
x^=ans,y^=ans;
int f1=find(x),f2=find(y);
if (size[f1]>size[f2]) swap(x,y),swap(f1,f2);
f[x][0]=y;rebuild(x,y);add(x,y);add(y,x);
ufs[f1]=f2;size[f2]+=size[f1];
}
}
}

总结