银河英雄传说 ← 带权并查集

【问题描述】
有一个划分为N列的星际战场，各列依次编号为1,2,…,N。
有N艘战舰，也依次编号为1,2,…,N，其中第i号战舰处于第i列。
有T条指令，每条指令格式为以下两种之一：
1、M i j，表示让第i号战舰所在列的全部战舰保持原有顺序，接在第j号战舰所在列的尾部。
2、C i j，表示询问第i号战舰与第j号战舰当前是否处于同一列中，如果在同一列中，它们之间间隔了多少艘战舰。
现在需要你编写一个程序，处理一系列的指令。

【输入格式】
第一行包含整数T，表示共有T条指令。
接下来T行，每行一个指令，指令有两种形式：M i j或C i j。
其中M和C为大写字母表示指令类型，i和j为整数，表示指令涉及的战舰编号。

【输出格式】
你的程序应当依次对输入的每一条指令进行分析和处理：
如果是M i j形式，则表示舰队排列发生了变化，你的程序要注意到这一点，但是不要输出任何信息；
如果是C i j形式，你的程序要输出一行，仅包含一个整数，表示在同一列上，第i号战舰与第j号战舰之间布置的战舰数目，如果第i号战舰与第j号战舰当前不在同一列上，则输出-1。

【数据范围】
N≤30000，T≤500000

【输入样例】
4
M 2 3
C 1 2
M 2 4
C 4 2

【输出样例】
-1
1

【算法分析】
带权并查集与经典并查集相比，通常只定义find()函数，不定义merge()函数。
本题也是一道带权并查集的经典题目。其中：
● d[i]表示第i号战舰到头部战舰的距离。
● size[i]表示集合i的元素个数。
● 本代码中，char op; 与 cin>>op; 配合使用，否则用其他方式会出现莫名其妙的错误。
● 本代码中，先用语句 int fx=find(x); 及 int fy=find(y); 声明fx、fy 后，再执行 fx!=fy，不要直接比较 find(x)!=find(y)，否则在带权并查集代码中会出现莫名其妙的错误。虽然，它们看起来似乎等价。

【算法代码】

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;

const int maxn=300005;
int pre[maxn];
int s[maxn]; //s[i]:the number of elements in set i
int d[maxn]; //d[i]:the distance from ith warship to the head warship

int T;

int find(int x) {
	if(x!=pre[x]) {
		int t=pre[x];
		pre[x]=find(pre[x]);
		d[x]+=d[t]; //Some operation
	}
	return pre[x];
}

int main() {
	scanf("%d",&T);

	for(int i=1; i<=maxn; i++) {
		pre[i]=i;
		s[i]=1;
	}

	char op;
	int x, y;
	while(T--){
		cin>>op;
		scanf("%d%d",&x,&y);
		if(op=='M') {
			int fx=find(x);
			int fy=find(y);			
			pre[fx]=fy; //add x to y
			d[fx]=s[fy];
			s[fy]+=s[fx];
		} 
		else {
			int fx=find(x);
			int fy=find(y);
			if(fx!=fy) printf("-1\n");
			else printf("%d\n", max(0, abs(d[x]-d[y])-1));
		}
	}
}


/*
in1:
4
M 2 3
C 1 2
M 2 4
C 4 4

out1:
-1
0

in2:
4
M 2 3
C 1 2
M 2 4
C 4 2

out2：
-1
1
*/

【参考文献】
https://blog.csdn.net/weixin_47456072/article/details/109598757
http://www.cnblogs.com/donkey2603089141/p/11415486.html
https://www.acwing.com/problem/content/240/
https://www.acwing.com/solution/content/11000/
https://www.acwing.com/solution/content/1005/