【bzoj3123】【sdoi2013】【森林】【启发式合并+主席树】

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。之后的操作类似。 

 

题解：相比count on a tree 这道题只是多了一个启发式合并.

            保存一下每棵树的size,每次合并将size小的那棵树暴力重构即可.

            注意在更新fa数组的时候一定要更新到底.

            因为一个点合并之后的深度可能会变小,如果光更新到这个点深度会导致一些以前的信息没有删除.

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 100010
#define M 20000010
#define inf 1000000000
using namespace std;
int point[N],next[N<<1],cnt,ans,deep[N],fa[N][20],n,m,x,y,test,v[N],q,k,size[N];
int root[N],ls[M],rs[M],sum[M],sz,a,b,c,d,bl[N];
char ch[5];
struct use{int st,en;}e[N<<1];
int read(){
  int x(0);
  char ch=getchar();
  while (ch<'0'||ch>'9') ch=getchar();
  while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
  return x;
}
void add(int x,int y){
  next[++cnt]=point[x];point[x]=cnt;
  e[cnt].en=y;
}
void insert(int x,int &y,int l,int r,int v){
  y=++sz;sum[y]=sum[x]+1;
  if (l==r) return;
  ls[y]=ls[x];rs[y]=rs[x];
  int mid=(l+r)>>1;
  if (v<=mid) insert(ls[x],ls[y],l,mid,v);
  else insert(rs[x],rs[y],mid+1,r,v);
}
int lca(int x,int y){
  if (deep[x]<deep[y]) swap(x,y);int t=deep[x]-deep[y];
  for (int i=0;i<=18;i++) if (t&(1<<i)) x=fa[x][i];
  for (int i=18;i>=0;i--) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
  if (x==y) return x;else return fa[x][0];
}
int query(int a,int b,int c,int d,int l,int r,int k){
  if (l==r) return l;
  int mid=(l+r)>>1,temp=k-sum[ls[b]]-sum[ls[a]]+sum[ls[c]]+sum[ls[d]];
  if (sum[ls[b]]+sum[ls[a]]-sum[ls[c]]-sum[ls[d]]>=k) 
    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,temp);
}
void dfs(int x,int f,int rt){
  fa[x][0]=f;deep[x]=deep[f]+1;size[x]=1;bl[x]=rt;
  insert(root[f],root[x],1,inf,v[x]);
  for (int i=1;i<20;i++) fa[x][i]=fa[fa[x][i-1]][i-1];
  for (int i=point[x];i;i=next[i])
    if (e[i].en!=fa[x][0]){
      dfs(e[i].en,x,rt);size[x]+=size[e[i].en];
    }
}
int main(){
  test=read();deep[0]=-1;
  n=read();m=read();q=read();
  for (int i=1;i<=n;i++) v[i]=read();
  for (int i=1;i<=m;i++){
    x=read();y=read();
    add(x,y);add(y,x);
  }	
  for (int i=1;i<=n;i++)
    if (!fa[i][0]) dfs(i,0,i);
  for (int i=1;i<=q;i++){
    scanf("%s",ch);
    if (ch[0]=='Q'){
       x=read();y=read();k=read();
       x^=ans;y^=ans;k^=ans;
       c=lca(x,y);d=fa[c][0];
       ans=query(root[x],root[y],root[c],root[d],1,inf,k);
       printf("%d\n",ans);
    }
    else{
      x=read();y=read();
      x^=ans;y^=ans;
      add(x,y);add(y,x);
      if (size[bl[x]]>size[bl[y]]) swap(x,y);
      size[bl[y]]+=size[bl[x]];
	  dfs(x,y,bl[y]);
    }
  }
}