洛谷3302 [SDOI2013]森林（主席树+启发式合并）

qwq好题啊

其实看到题目的第一眼，我以为是LCT。
但是发现做起来很麻烦
qwq
首先 树上路径第$k$大的问题可以直接通过主席树来实现。

那么连边这个操作该怎么办呢？

qwq我们发现
题目保证连完边之后还是一个森林。

那么这时候 我们会发现最终的形态会是一棵树

那么可以考虑启发式合并。

对于每一次连边，我们重新遍历小的联通块，更新主席树数组和$f$数组。

void change(int x,int fa)
{
	deep[x]=deep[fa]+1;
	root[x]=modify(root[fa],1,sz,val[x]);
	f[x][0]=fa;
	for (int i=1;i<=20;i++) f[x][i]=f[f[x][i-1]][i-1];
	for (int i=point[x];i;i=nxt[i])
	{
		int p = to[i];
		if (p==fa) continue;
		change(p,x);
	}
} 

qwq
那么这个题也就解决啦！
复杂度是$O(nlo{g}^{2}n)$

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<cmath>
#include<map>
#include<set>
#define pb push_back
#define mk make_pair
#define ll long long
#define lson ch[x][0]
#define rson ch[x][1]

using namespace std;

inline int read()
{
  int x=0,f=1;char ch=getchar();
  while (!isdigit(ch)) {if (ch=='-') f=-1;ch=getchar();}
  while (isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
  return x*f;
}

const int maxn = 3e5+1e2;
const int maxm = 2*maxn;

int point[maxn],nxt[maxm],to[maxm];
int cnt,n,m,q;
int f[maxn][21];
int root[maxn];
int deep[maxn],size[maxn];
int val[maxn];

struct Node{
	int l,r,val;
};

Node t[maxn*40];

void addedge(int x,int y)
{
	nxt[++cnt]=point[x];
	to[cnt]=y;
	point[x]=cnt;
}

void init()
{
	for (int j=1;j<=20;j++)
	  for (int i=1;i<=n;i++)
	    f[i][j]=f[f[i][j-1]][j-1];
}

int go_up(int x,int d)
{
	for (int i=0;i<=20;i++)
	{
		if ((1<<i) & d) x=f[x][i];
	}
	return x;
}

int lca(int x,int y)
{
	if (deep[x]>deep[y]) x=go_up(x,deep[x]-deep[y]);
	else y = go_up(y,deep[y]-deep[x]);
	if (x==y) return x;
	for (int i=20;i>=0;i--)
	{
		if (f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

int tot;

void up(int root)
{
	t[root].val=t[t[root].l].val+t[t[root].r].val;
}

int modify(int pre,int l,int r,int x)
{
	int now = ++tot;
	if(l==r)
	{
		t[now].val=t[pre].val+1;
		return now;
	}
	int mid = l+r >> 1; 
	if (x<=mid)
	{
		t[now].r=t[pre].r;
		t[now].l=modify(t[pre].l,l,mid,x);
	}
	else
	{
		t[now].l=t[pre].l;
		t[now].r=modify(t[pre].r,mid+1,r,x);
	}
	up(now);
	return now; 
}

vector<int> v;
int sz;

int getnum(int val)
{
	return lower_bound(v.begin(),v.end(),val)-v.begin()+1;
}

void dfs(int x,int fa,int dep)
{
	size[x]=1;
	deep[x]=dep;
	root[x]=modify(root[fa],1,sz,val[x]);
	for (int i=point[x];i;i=nxt[i])
	{
		int p = to[i];
		if (p==fa) continue;
		dfs(p,x,dep+1);
		f[p][0]=x; 
		size[x]+=size[p];
	}
}

int query(int root1,int root2,int root3,int root4,int l,int r,int k)
{
	if (l==r) return v[l-1];
	int mid = l+r >> 1;
	int vall = t[t[root1].l].val+t[t[root2].l].val-t[t[root3].l].val-t[t[root4].l].val;
	if (vall>=k) return query(t[root1].l,t[root2].l,t[root3].l,t[root4].l,l,mid,k);
	else return query(t[root1].r,t[root2].r,t[root3].r,t[root4].r,mid+1,r,k-vall);
}

void change(int x,int fa)
{
	deep[x]=deep[fa]+1;
	root[x]=modify(root[fa],1,sz,val[x]);
	f[x][0]=fa;
	for (int i=1;i<=20;i++) f[x][i]=f[f[x][i-1]][i-1];
	for (int i=point[x];i;i=nxt[i])
	{
		int p = to[i];
		if (p==fa) continue;
		change(p,x);
	}
} 

void clear()
{
	tot=0;
	memset(point,0,sizeof(point));
	cnt=0;
	memset(deep,0,sizeof(deep));
	memset(size,0,sizeof(size));
	v.clear();
	memset(f,0,sizeof(f));
	memset(root,0,sizeof(root));
}

int T;

int main()
{
  cin>>T;
  n=read();m=read();q=read();
  for (int i=1;i<=n;i++)
  {
  	 val[i]=read();
  	 v.pb(val[i]);
  }
  sort(v.begin(),v.end());
  sz = unique(v.begin(),v.end())-v.begin();
  v.resize(sz); 
  for (int i=1;i<=n;i++) val[i]=getnum(val[i]);
  for (int i=1;i<=m;i++)
  {
  	 int x=read(),y=read();
  	 addedge(x,y);
  	 addedge(y,x);
  }
  for (int i=1;i<=n;i++)
  {
  	if (!deep[i]) dfs(i,0,1);
  }
  init();
  int lastans=0;
  for (int i=1;i<=q;i++)
  {
  	 char s[10];
  	 scanf("%s",s+1);
  	 if (s[1]=='Q')
  	 {
  	 	int x=read(),y=read(),k=read();
  	 	x^=lastans;
  	 	y^=lastans;
  	 	k^=lastans;
  	 	int l = lca(x,y);
  	 	lastans = query(root[x],root[y],root[l],root[f[l][0]],1,sz,k);
  	 	cout<<lastans<<"\n";
	 }
	 else
	 {
	 	int x=read(),y=read();
	 	x^=lastans;
	 	y^=lastans;
	 	int xx = go_up(x,deep[x]-1);
	 	int yy = go_up(y,deep[y]-1);
	    if (size[xx]>size[yy]) swap(xx,yy),swap(x,y);
	    size[yy]+=size[xx];
	    addedge(x,y);
	    addedge(y,x);
	    change(x,y);
	 }
  }
  return 0;
}