题意:
有n列战舰,有两种操作,第一种,把一列战舰放在另一列的后面,也就是合并操作,第二种是查询操作,问两个战舰是不是在同一列,如果是的话,就输出它们之间隔了多少个战舰,如果不是就输出-1.
思路:
用一个数组表示是不是在一个集合里,然后用另一个数组d表示每一列的大小,还有一个数组siz表示边的权值。
d是到祖先的距离,siz是这列的大小。如果在同一列里面,就直接输出d[x]-dy[y]的绝对值减一。
#pragma warning(disable:4996)
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
#include<vector>
#include<map>
#include<string>
#include<set>
#include<queue>
using namespace std;
typedef long long ll;
int fa[30005], d[30005], siz[30005];
int find(int x)
{
if (x != fa[x])
{
int t = fa[x];
fa[x] = find(fa[x]);
d[x] += d[t];//到祖先的距离
}
return fa[x];
}
int main()
{
int i, j, fx, fy, x, y,T;
char s[20];
for (i = 1; i <= 30000; i++)
{
fa[i] = i; siz[i] = 1;
}
scanf("%d", &T);
while (T--)
{
scanf("%s%d%d", s,&x,&y);
fx = find(x); fy = find(y);
if (s[0] == 'M')
{
fa[fx] = fy;
d[fx] = siz[fy];//fx的祖先改为了fy
siz[fy] += siz[fx];//fy的大小
}
else
{
if (fx == fy)printf("%d\n", abs(d[x] - d[y])-1);
else printf("-1\n");
}
}
return 0;
}