题目:
题解:
一个是链上求第k大,另一个是连接一条边,强制在线
这样不是连通块内的第k大我们考虑权值线段树,连接一条边就是线段树的合并,对于链上的问题,我们需要采用可持久化的线段树-主席树,而且是树上主席树,即给每个点建一棵主席树,维护到根节点的所有点,这样答案就是sum(a)+sum(b)−sum(lca(a,b))−sum(fa(lca(a,b)))
还要连接一条边的话就是主席树的合并,合并要注意的就是合并之后原来的爸爸可能变成了儿子,即父子关系会发生变化,为了避免混乱,f数组要全部清空;要使用启发式合并,size小的放到size大的里
还要小技巧就是一棵树的size可以用并查集的根节点来维护,这样不用每次都更新
代码:
#include <map>
#include <cstdio>
#include <algorithm>
using namespace std;
const int sz=18;
const int N=80005;
int mi[sz+5],f[N][sz+5],tot,nxt[N*2],point[N],s,v[N*2],sum[N*200],ls[N*200],rs[N*200],h[N],zs,root[N];
int size[N],a[N],real[N],fa[N];map<int,int>has;
bool vis[N];
void addline(int x,int y)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void insert(int &now,int l,int r,int x)
{
sum[++zs]=sum[now]+1; ls[zs]=ls[now]; rs[zs]=rs[now]; now=zs;
if (l==r) return;
int mid=(l+r)>>1;
if (x<=mid) insert(ls[now],l,mid,x);
else insert(rs[now],mid+1,r,x);
}
void dfs(int x,int fa,int id)
{
vis[x]=1;h[x]=h[fa]+1;
root[x]=root[fa];if (id==0) size[x]=1;
insert(root[x],1,s,a[x]);
for (int i=1;i<sz;i++) 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,dfs(v[i],x,id);
if (id==0) size[x]+=size[v[i]];
}
}
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<sz;i++)
if (k>>i&1) x=f[x][i];
if (x==y) return x;
for (int i=sz-1;i>=0;i--)
if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
int qurry(int l,int r,int k,int x,int y,int old,int oldf)
{
int mid=(l+r)>>1;
if (l==r) return l;
int tmp=sum[ls[x]]+sum[ls[y]]-sum[ls[old]]-sum[ls[oldf]];
if (tmp>=k) return qurry(l,mid,k,ls[x],ls[y],ls[old],ls[oldf]);
else return qurry(mid+1,r,k-tmp,rs[x],rs[y],rs[old],rs[oldf]);
}
int find(int x){if (fa[x]!=x) fa[x]=find(fa[x]);return fa[x];}
int main()
{
mi[0]=1;for (int i=1;i<sz;i++) mi[i]=mi[i-1]*2;
int T,n,m,q;scanf("%d",&T);
scanf("%d%d%d",&n,&m,&q);
for (int i=1;i<=n;i++) scanf("%d",&a[i]),h[i]=a[i],fa[i]=i;
sort(h+1,h+n+1);
s=unique(h+1,h+n+1)-h-1;
for (int i=1;i<=s;i++) has[h[i]]=i,real[i]=h[i];
for (int i=1;i<=n;i++) a[i]=has[a[i]];
for (int i=1;i<=m;i++)
{
int x,y;scanf("%d%d",&x,&y);
addline(x,y);x=find(x);y=find(y);
fa[x]=y;
}
for (int i=1;i<=n;i++)
if (!vis[i]) dfs(i,0,0);
int ans=0;
while (q--)
{
int x,y,k;char st[5];scanf("%s",st);
if (st[0]=='L')
{
scanf("%d%d",&x,&y); x^=ans; y^=ans;
int fx=find(x),fy=find(y);
if (size[fx]>size[fy]) swap(x,y),swap(fx,fy);
f[x][0]=y; dfs(x,y,1); addline(x,y);
size[fy]+=size[fx];fa[fx]=fy;
}else
{
scanf("%d%d%d",&x,&y,&k);x^=ans; y^=ans; k^=ans;
int old=lca(x,y),oldf=f[old][0];
ans=real[qurry(1,s,k,root[x],root[y],root[old],root[oldf])];
printf("%d\n",ans);
}
}
}