树上主席树+启发式合并
链接:[P3302 SDOI2013]森林 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
题意:给定一座森林,有多次操作,可以查询 x 到 y 路径上第 k 小,也可以将两棵树连接成一棵。
题解:首先区间第 k 小,主席树维护。树上第 k 小同理主席树,继承时选择从父亲节点继承。查询时查同时查 4 个点 x , y , l c a , f a [ l c a ] x,y,lca,fa[lca] x,y,lca,fa[lca]。现在多了个连接操作,由于最多连接 n-1 次后最后成一棵树,因此考虑启发式合并,每次合并将小的树暴力拆掉接到大的树下,均摊复杂度每次为 log 级别,每个点会重建倍增数组也是复杂度 log,整体修改为 O ( n l o g 2 n ) O(nlog^2n) O(nlog2n)。用并查集维护下树的集合方便修改和求树大小。
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include<iostream>
#include<algorithm>
#include<vector>
#include<cstring>
using namespace std;
const int maxn=1e5+5;
char op;
int a[maxn],b[maxn],n,m,q,mx,u,v,w,ans;
int dep[maxn],fa[maxn][18],lg[maxn],ft[maxn],sz[maxn];
int ls[maxn<<7],rs[maxn<<7],sum[maxn<<7],rt[maxn],now;
vector<int>ed[maxn];
int find(int x)
{
return ft[x]==x?x:ft[x]=find(ft[x]);
}
void update(int u,int&v,int l,int r,int pos)
{
v=++now,ls[v]=ls[u],rs[v]=rs[u],sum[v]=sum[u]+1;
if(l==r)return;
int mid=l+r>>1;
if(pos<=mid)update(ls[u],ls[v],l,mid,pos);
else update(rs[u],rs[v],mid+1,r,pos);
return;
}
int query(int u1,int u2,int v1,int v2,int l,int r,int k)
{
if(l==r)return b[l];
int mid=l+r>>1,num=sum[ls[v1]]+sum[ls[v2]]-sum[ls[u1]]-sum[ls[u2]];
if(num>=k)return query(ls[u1],ls[u2],ls[v1],ls[v2],l,mid,k);
else return query(rs[u1],rs[u2],rs[v1],rs[v2],mid+1,r,k-num);
}
int LCA(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
while(dep[x]>dep[y])
{
x=fa[x][lg[dep[x]-dep[y]]-1];
}
if(x==y)return x;
for(int i=17;i>=0;i--)
{
if(fa[x][i]!=fa[y][i])
{
x=fa[x][i],y=fa[y][i];
}
}
return fa[x][0];
}
void dfs(int x,int f,int t)
{
dep[x]=dep[f]+1,fa[x][0]=f,sz[t]++,ft[x]=t;
update(rt[f],rt[x],1,mx,a[x]);
for(int i=1;i<=17;i++)
{
fa[x][i]=fa[fa[x][i-1]][i-1];
}
for(auto y:ed[x])
{
if(y!=f)dfs(y,x,t);
}
return;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
int T; cin>>T;
for(int i=1;i<=1e5;i++)lg[i]=lg[i-1]+(!(i&i-1));
cin>>n>>m>>q;
for(int i=1;i<=n;i++)cin>>a[i],b[i]=a[i],ft[i]=i;
sort(b+1,b+1+n); mx=unique(b+1,b+1+n)-b-1;
for(int i=1;i<=n;i++)a[i]=lower_bound(b+1,b+1+mx,a[i])-b;
for(int i=1;i<=m;i++)
{
cin>>u>>v;
ed[u].push_back(v);
ed[v].push_back(u);
}
for(int i=1;i<=n;i++)if(ft[i]==i)dfs(i,0,i);
for(int i=1;i<=q;i++)
{
cin>>op;
if(op=='Q')
{
cin>>u>>v>>w;
u^=ans,v^=ans,w^=ans;
int lca=LCA(u,v);
ans=query(rt[lca],rt[fa[lca][0]],rt[u],rt[v],1,mx,w);
cout<<ans<<"\n";
}
else
{
cin>>u>>v;
u^=ans,v^=ans;
int x=find(u),y=find(v);
if(sz[x]<sz[y])swap(x,y),swap(u,v);
dfs(v,u,x),sz[y]=0;
ed[v].push_back(u);
ed[u].push_back(v);
}
}
return 0;
}