[Acwing ] 238. 银河英雄传说 (边带权并查集)

题目链接：238.银河英雄传说

题解

本题的难点在于查询同一集合的两艘战舰间隔多少战舰的问题。

我们可以利用前缀和的思想
维护一个d[x]：x到f[x](x所属集合终结点)的距离。
维护一个集合大小sz[x]：集合x的点数。

每当我们要将a集合合并到b集合上时，很明显b集合上的d[x]值是不会发生改变的，影响的是a集合上的值，需要在a集合上每个点的d[x]+sz[b]，也就是集合a的点到b集合终结点的距离。

但注意其实我们在路径压缩时，会不断的访问父节点、父节点的父节点以至于找到集合最终结点，所以可以得到一个递推式：d[x]=d[x]+d[f[x]]，d[f[x]]=d[f[x]]+d[f[f[x]]]…直到x==f[x]终止
因此只需在a集合终结点的d[x]变为sz[b]即可，在路径压缩时，把距父结点的权值不断由上到下叠加得到。

最后的答案就是：abs(d[a]-d[b])-1

代码

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<bitset>
#include<cassert>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<ctime>
#include<deque>
#include<iomanip>
#include<list>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<unordered_set>
#include<unordered_map>
using namespace std;
//extern "C"{void *__dso_handle=0;}
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define pii pair<int,int>
#define lowbit(x) x&-x

const double PI=acos(-1.0);
const double eps=1e-6;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
const int maxn=1e5+10;
const int maxm=100+10;
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

int d[maxn],sz[maxn],f[maxn];
int find(int x)
{
	if(x!=f[x])
	{
		int rt=find(f[x]);
		d[x]+=d[f[x]];
		f[x]=rt;
	}
	return f[x];
}

int main()
{
	for(int i=1;i<maxn;i++)
	{
		f[i]=i;
		sz[i]=1;
	}
	int t;
	scanf("%d",&t);
	while (t--) {
		char op[2]; scanf("%s",op);
		if(op[0]=='M')
		{
			int a,b; scanf("%d%d",&a,&b);
			int fa=find(a),fb=find(b);
			d[fa]=sz[fb];
			sz[fb]+=sz[fa];
			f[fa]=fb;
		}
		else {
			int a,b; scanf("%d%d",&a,&b);
			int fa=find(a),fb=find(b);
			if(fa!=fb) printf("-1\n");
			else
			{
				printf("%d\n",max(0,abs(d[a]-d[b])-1));
			}
		}
	}
}