[主席树 启发式合并] BZOJ 3123 [Sdoi2013]森林

题目大意：带link的森林中 树上路径第k大

本来一看LCT 后来看第k大 主席树啦

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#define V G[p].v
#define cl(x) memset(x,0,sizeof(x))
using namespace std;

inline char nc(){
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x){
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

inline void read(char &x){
	for (x=nc();x!='Q' && x!='L';x=nc());
}

const int N=80005;
const int M=10000005;

int sx[N],icnt;

inline int Bin(int x){
	return lower_bound(sx+1,sx+icnt+1,x)-sx;
}

int n,m,Q,val[N];

int ncnt;
int root[N],ls[M],rs[M],sum[M];

int Stk[M],pnt;
inline void Init(){ pnt=0; for (int i=M-1;i;i--) Stk[++pnt]=i; }
inline int New(){ return Stk[pnt--]; }
inline void Del(int x){ Stk[++pnt]=x; }

void Clear(int &x){
	if (!x) return;
	Clear(ls[x]); Clear(rs[x]);
	Del(x); x=0; sum[x]=0;
}

void Modify(int &y,int x,int l,int r,int t){
	y=++ncnt; int mid=(l+r)>>1;
	if (l==r) { sum[y]=sum[x]+1; return; }
	if (t<=mid)
		rs[y]=rs[x],Modify(ls[y],ls[x],l,mid,t),sum[y]=sum[ls[y]]+sum[rs[y]];
	else
		ls[y]=ls[x],Modify(rs[y],rs[x],mid+1,r,t),sum[y]=sum[ls[y]]+sum[rs[y]];
}

int Query(int a,int b,int c,int d,int l,int r,int K){
	if (l==r) return sx[l];
	int mid=(l+r)>>1,tmp=sum[ls[a]]+sum[ls[b]]-sum[ls[c]]-sum[ls[d]];
	if (K<=tmp)
		return Query(ls[a],ls[b],ls[c],ls[d],l,mid,K);
	else
		return Query(rs[a],rs[b],rs[c],rs[d],mid+1,r,K-tmp);
}

struct edge{
	int u,v;
	int next;
};

edge G[2*N];
int head[N],inum;

inline void add(int u,int v,int p){
	G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int fat[N][21],depth[N],vst[N];

inline void dfs(int u,int fa){
	vst[u]=1; fat[u][0]=fa; depth[u]=depth[fa]+1;
	for (int k=1;k<=20;k++) fat[u][k]=fat[fat[u][k-1]][k-1];
	Modify(root[u],root[fa],1,icnt,val[u]);
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
			dfs(V,u);
}

inline int LCA(int u,int v){
	if (depth[u]<depth[v]) swap(u,v);
	for (int k=20;~k;k--)
		if ((depth[u]-depth[v])&(1<<k))
			u=fat[u][k];
	if (u==v) return u;
	for (int k=20;~k;k--)
		if (fat[u][k]!=fat[v][k])
			u=fat[u][k],v=fat[v][k];
	return fat[u][0];
}

inline void Clean(int u,int fa){
	Clear(root[u]);
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa)
			Clean(V,u);
}

namespace UQS{
	int fat[N],sum[N];
	inline void init(int n){
		for (int i=1;i<=n;i++) fat[i]=i,sum[i]=1;
	}
	inline int Fat(int u){
		return u==fat[u]?u:fat[u]=Fat(fat[u]);
	}
	inline void Union(int x,int y){
		int fx=Fat(x),fy=Fat(y);
		if (fx!=fy)
			fat[fx]=fy,sum[fy]+=sum[fx];
	}
	inline int Sum(int u){
		return sum[Fat(u)];
	}
}

inline void Link(int u,int v){
	if (UQS::Sum(u)>UQS::Sum(v)) swap(u,v);
	Clean(u,0);
	dfs(u,v);
	add(u,v,++inum); add(v,u,++inum); 
	UQS::Union(u,v);
}

int main()
{
	int iu,iv,k,last=0,lca;
	char order;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	Init();
	read(n); read(n); read(m); read(Q);
	UQS::init(n);
	for (int i=1;i<=n;i++) read(val[i]),sx[++icnt]=val[i];
	sort(sx+1,sx+icnt+1); icnt=unique(sx+1,sx+icnt+1)-sx-1;
	for (int i=1;i<=n;i++) val[i]=Bin(val[i]);
	for (int i=1;i<=m;i++) read(iu),read(iv),add(iu,iv,++inum),add(iv,iu,++inum),UQS::Union(iu,iv);
	for (int i=1;i<=n;i++) if (!vst[i]) dfs(i,0);
	while (Q--)
	{
		read(order);
		if (order=='L')
		{
			read(iu); read(iv); iu^=last; iv^=last; 
			Link(iu,iv);
		}
		else
		{
			read(iu); read(iv); read(k); iu^=last; iv^=last; k^=last; 
			lca=LCA(iu,iv);
			last=Query(root[iu],root[iv],root[lca],root[fat[lca][0]],1,icnt,k);
			printf("%d\n",last);
		}
	}
}