边带权并查集 acwing 238

题意：

有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;
}

 

来源：CSDN

作者：sunnyorrainy

链接：https://blog.csdn.net/qq_43552826/article/details/104181587