【bzoj3123】[Sdoi2013]森林 倍增LCA+主席树+启发式合并

题目描述



输入

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

输出

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

样例输入

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

题解

倍增LCA+主席树+启发式合并

如果没有连边操作，那么本题同 bzoj2588 。

好在本题的n只有80000，所以我们可以使用一些高（qi）端（ji）姿（yin）势（qiao）来解决。

由于只有连边没有删边，所以可以使用启发式合并，暴力将较小的树连到较大的树上，从连接点开始再dfs一遍更新fa和deep。

同时需要记录每棵树的大小，相当于记录每个点的树根。

然后就是建树，求LCA，求出答案。

这里需要注意的一点是，对于不同的倍增LCA的写法，如果写法中利用到f[x][...]=0，那么务必在连边时将原来的f数组清空，否则当原深度大于新深度时会WA->RE。

#include <cstdio>
#include <algorithm>
#define N 80010
using namespace std;
int n , w
, a
, ref
, head
, to[N << 1] , next[N << 1] , cnt , fa
[20] , deep
, log
, bl
, si
;
int ls[N << 8] , rs[N << 8] , sum[N << 8] , root
, tot;
char str[5];
void add(int x , int y)
{
to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt;
}
void insert(int p , int l , int r , int x , int &y)
{
if(!y) y = ++tot;
sum[y] = sum[x] + 1;
if(l == r) return;
int mid = (l + r) >> 1;
if(p <= mid) rs[y] = rs[x] , insert(p , l , mid , ls[x] , ls[y]);
else ls[y] = ls[x] , insert(p , mid + 1 , r , rs[x] , rs[y]);
}
int query(int p , int l , int r , int a , int b , int c , int d)
{
if(l == r) return ref[l];
int mid = (l + r) >> 1;
if(sum[ls[a]] + sum[ls[b]] - sum[ls[c]] - sum[ls[d]] >= p) return query(p , l , mid , ls[a] , ls[b] , ls[c] , ls[d]);
else return query(p - sum[ls[a]] - sum[ls[b]] + sum[ls[c]] + sum[ls[d]] , mid + 1 , r , rs[a] , rs[b] , rs[c] , rs[d]);
}
void dfs(int x , int r)
{
int i;
bl[x] = r , si[r] ++ ;
insert(w[x] , 1 , n , root[fa[x][0]] , root[x]);
for(i = 1 ; i <= log[deep[x]] ; i ++ ) fa[x][i] = fa[fa[x][i - 1]][i - 1];
for(i = head[x] ; i ; i = next[i])
if(to[i] != fa[x][0])
fa[to[i]][0] = x , deep[to[i]] = deep[x] + 1 , dfs(to[i] , r);
}
int lca(int x , int y)
{
int i;
if(deep[x] < deep[y]) swap(x , y);
for(i = log[deep[x] - deep[y]] ; i >= 0 ; i -- )
if(deep[x] - deep[y] >= (1 << i))
x = fa[x][i];
for(i = log[deep[x]] ; i >= 0 ; i -- )
if(deep[x] >= (1 << i) && fa[x][i] != fa[y][i])
x = fa[x][i] , y = fa[y][i];
return x == y ? x : fa[x][0];
}
int main()
{
int m , q , i , t , x , y , z , last = 0;
scanf("%*d%d%d%d" , &n , &m , &q);
for(i = 1 ; i <= n ; i ++ ) scanf("%d" , &w[i]) , a[i] = w[i];
sort(a + 1 , a + n + 1);
for(i = 1 ; i <= n ; i ++ ) t = w[i] , w[i] = lower_bound(a + 1 , a + n + 1 , w[i]) - a , ref[w[i]] = t;
for(i = 1 ; i <= m ; i ++ ) scanf("%d%d" , &x , &y) , add(x , y) , add(y , x);
log[0] = -1;
for(i = 2 ; i <= n ; i ++ ) log[i] = log[i >> 1] + 1;
for(i = 1 ; i <= n ; i ++ ) if(!fa[i][0]) dfs(i , i);
while(q -- )
{
scanf("%s%d%d" , str , &x , &y) , x ^= last , y ^= last;
if(str[0] == 'Q')
{
scanf("%d" , &z) , z ^= last , t = lca(x , y);
printf("%d\n" , last = query(z , 1 , n , root[x] , root[y] , root[t] , root[fa[t][0]]));
}
else
{
if(si[bl[x]] < si[bl[y]]) swap(x , y);
fa[y][0] = x , deep[y] = deep[x] + 1 , dfs(y , bl[x]) , add(x , y) , add(y , x);
}
}
return 0;
}