【Ybtoj 第10章例3】银河英雄传说【并查集】

---

解题思路

这道题要引入一种并查集的拓展用法：边带权的并查集。

这里的边权可以表示很多信息，例如某个节点和他的父亲节点中的距离，或是数值之和异或之和等。。大体做法与普通并查集相似，但在合并集合和路径压缩的时候要更新边权。

回到本题，我们可以用并查集维护战舰是否在同一列，以每列的第一艘战舰作为集合代表，用$dis$记录边权，$di{s}_x$表示编号为x的战舰到编号为 $f{a}_x$的战舰之间有几艘战舰（不包括x），同时维护一个$num$数组，$nu{m}_x$表示以x为代表元素的集合有多少艘战舰。
对于查询操作，我们先判断$x，y$是否在同一列中（及所在集合的代表元素是否相同），再分别求出他们与代表元素之间有多少艘战舰，相减所得的差即为所求的答案。

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<bitset>
using namespace std;

int  T,x,y,fa[30010],num[30010],dis[30010];
char c;

int find(int x)
{
	if(fa[x]==x)return fa[x];	
	int v=find(fa[x]);
	dis[x]=dis[fa[x]]+dis[x];//自己到新爸爸的距离，等于自己的原爸爸到新爸爸的距离，加上自己到原爸爸的距离
	return fa[x]=v;
}

int main(){
	scanf("%d",&T);
	for(int i=1;i<=30009;i++)
		fa[i]=i,num[i]=1;
	while(T--)
	{
		cin>>c;
		scanf("%d%d",&x,&y);
		int xx=find(x),yy=find(y);
		if(c=='M')
		{
			fa[xx]=yy;
			dis[xx]=num[yy];//加上到新爸爸的距离（新增距离相当于新爸爸所在集合的长度）
			num[yy]+=num[xx];//加上新合并进来的长度
			num[xx]=0;//不存在了，被合并了
		}
		if(c=='C')
		{
			if(xx!=yy) printf("-1\n");
			else printf("%d\n",abs(dis[y]-dis[x])-1);//两飞船间距离
		}
	}
}