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

3123: [Sdoi2013]森林

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3563  Solved: 1038
[Submit][Status][Discuss]

Description

Input

第一行包含一个正整数testcase，表示当前测试数据的测试点编号。保证1≤testcase≤20。 
第二行包含三个整数N，M，T，分别表示节点数、初始边数、操作数。第三行包含N个非负整数表示 N个节点上的权值。 
 接下来 M行，每行包含两个整数x和 y，表示初始的时候，点x和点y 之间有一条无向边, 接下来 T行，每行描述一个操作，格式为“Q x y k”或者“L x y ”，其含义见题目描述部分。

Output

对于每一个第一类操作，输出一个非负整数表示答案。 

Sample Input

1
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3 Q 3 5 1
Q 10 0 0
L 5 4
L 3 2 L 0 7
Q 9 2 5 Q 6 1 6

Sample Output

2
2
1
4
2

HINT

对于第一个操作 Q 8 7 3，此时 lastans=0，所以真实操作为Q 8^0 7^0 3^0，也即Q 8 7 3。点8到点7的路径上一共有5个点，其权值为4 1 1 2 4。这些权值中，第三小的为 2，输出 2，lastans变为2。对于第二个操作 Q 3 5 1 ，此时lastans=2，所以真实操作为Q 3^2 5^2 1^2 ，也即Q 1 7 3。点1到点7的路径上一共有4个点，其权值为 1 1 2 4 。这些权值中，第三小的为2，输出2，lastans变为 2。之后的操作类似。 

最开始看testcase以为多组数据=_=

最后开心地一A啦啦啦啦

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>
using namespace std;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
void print(int x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100,M=40001000;

int ecnt,last[N];
struct EDGE{int to,nt;}e[N<<1];
inline void add(int u,int v)
{e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}

int root[N],maxn,cnt;

int V[N],ref[N];

struct president_tree{int ls,rs,w;}tr[M];

void insert(int &k,int x,int l,int r,int val)
{
	k=++cnt;
	tr[k].w=tr[x].w+1;tr[k].ls=tr[x].ls;tr[k].rs=tr[x].rs;
	if(l==r)return ;int mid=(l+r)>>1;
	val<=mid?insert(tr[k].ls,tr[x].ls,l,mid,val):insert(tr[k].rs,tr[x].rs,mid+1,r,val);
}

int fa[N][18],size[N],d[N],an[N];

int st[N],top;

void dfs(int u,int ar)
{
	st[++top]=u;d[u]=d[fa[u][0]]+1;size[u]=1;an[u]=ar;
	insert(root[u],root[fa[u][0]],1,maxn,V[u]);
	for(int i=last[u];i;i=e[i].nt)
	{
		if(e[i].to==fa[u][0])continue;
		fa[e[i].to][0]=u;dfs(e[i].to,ar);size[u]+=size[e[i].to];
	}
}

inline void lcainit()
{
	register int i,j;
	for(j=1;j<=17;++j)for(i=1;i<=top;++i)fa[st[i]][j]=fa[fa[st[i]][j-1]][j-1];
}

inline int getlca(int u,int v)
{
	register int i,len;
	if(d[u]<d[v])swap(u,v);
	len=d[u]-d[v];
	for(i=0;(1<<i)<=len;++i)if((1<<i)&len)u=fa[u][i];
	if(u==v)return u;
	for(i=17;i>=0;--i)if(fa[u][i]^fa[v][i])u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}

int query(int u,int v,int k)
{
	register int l=1,r=maxn,lca,falca,sum,mid;
	lca=getlca(u,v);falca=fa[lca][0];
	u=root[u];v=root[v];lca=root[lca];falca=root[falca];
	while(l<r)
	{
		sum=0;
		sum+=tr[tr[u].ls].w+tr[tr[v].ls].w;
		sum-=tr[tr[lca].ls].w+tr[tr[falca].ls].w;
		mid=(l+r)>>1;
		if(k<=sum)
		{
			u=tr[u].ls;v=tr[v].ls;
			lca=tr[lca].ls;falca=tr[falca].ls;
			r=mid;
		}
		else
		{
			u=tr[u].rs;v=tr[v].rs;
			lca=tr[lca].rs;falca=tr[falca].rs;
			k-=sum;l=mid+1;
		}
	}
	return l;
}

void link(int u,int v)
{
	register int x=an[u],y=an[v];
	add(u,v);add(v,u);
	if(size[x]>size[y])swap(u,v),swap(x,y);
	fa[u][0]=v;dfs(u,y);
	lcainit();top=0;
}

struct P{int val,pos;friend bool operator <(const P &x,const P &y){return x.val<y.val;}}p[N];

int main()
{
	register int n,m,Q,i,k,u,v,ans;
	char opt[2];
	Q=read();n=read();m=read();Q=read();ans=0;
	for(i=1;i<=n;++i){p[i].val=read();p[i].pos=i;}
	sort(p+1,p+1+n);
	k=1;ref[1]=p[1].val;V[p[1].pos]=1;
	for(i=2;i<=n;++i){if(p[i].val^p[k].val)ref[++k]=p[i].val;V[p[i].pos]=k;}
	maxn=k;
	for(i=1;i<=m;++i){u=read();v=read();add(u,v);add(v,u);}
	for(i=1;i<=n;++i)if(!fa[i][0]){dfs(i,i);lcainit();top=0;}
	while(Q--)
	{
		scanf("%s",opt);u=read()^ans;v=read()^ans;
		switch(opt[0])
		{
			case 'Q':k=read()^ans;print(ans=ref[query(u,v,k)]);puts("");break;
			case 'L':link(u,v);break;
			default:break;
		}
	}
	return 0;
}
/*
1
8 4 8
1 1 2 2 3 3 4 4
4 7
1 8
2 4
2 1
Q 8 7 3 Q 3 5 1
Q 10 0 0
L 5 4
L 3 2 L 0 7
Q 9 2 5 Q 6 1 6

2
2
1
4
2
*/