简化题意:
有 30000 个元素,初始时每个元素以单独的队列形式存在,支持一下两种操作:
1.动态合并两条队列,将 x 元素所在队列首合并在 y 元素所在队列尾;
2.查询 x 与 y 是否在同一条队列中,若是,查询 x 与 y 间隔元素数量。
共 500,000 次操作。
题解:
在父亲节点维护的时候记录每个并查集端点的元素大小,顺便维护一下就好了。
#include <bits/stdc++.h>
using namespace std;
int n;
int fa[30010], dis[30010], mx[30010];
int find(int x){
if(fa[x] == x) return x;
int t = find(fa[x]);
dis[x] += dis[fa[x]];
fa[x] = t;
mx[fa[x]] = max(mx[fa[x]], dis[x]);
return fa[x];
}
int main(){
scanf("%d", &n);
for(int i = 1; i <= 30000; i ++) fa[i] = i;
for(int i = 1; i <= n; i ++){
char ch[3];
int a, b;
scanf("%s%d%d", ch+1, &a, &b);
if(ch[1] == 'M'){
if(find(a) != find(b)){
dis[fa[b]] = mx[fa[a]]+1;
mx[fa[a]] += mx[fa[b]]+1;
fa[fa[b]] = fa[a];
}
}else{
if(find(a) != find(b)){
printf("-1\n");
}else printf("%d\n", abs(dis[b]-dis[a])-1);
}
}
return 0;
}