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
;
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
,maxn,cnt;

int V
,ref
;

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
[18],size
,d
,an
;

int st
,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
;

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
*/