银河英雄传说【并查集】

银河英雄传说

题目链接

https://www.acwing.com/problem/content/240/

题目描述

思路

维护并记录每个集合的长度，在新加入结点的时候，新节点距离根节点的距离就是该集合的长度 - 1

并查集可以用于维护具有传递性关系的作用，每个集合的大小，绑定到根结点上，每个点到根结点的距离，绑定到每个元素的结点上

• 1、size[x]表示集合的大小，d[x]表示x到p[x]的距离

• 2、将第a列的船接在第b列的末尾，如图所示，相当于让每个点到pb的距离都加上size[pb]，由于存储并查集中存在路径压缩的效果，因此只需要将pa到pb的距离加上size[pb]即可，即d[pa] = size[pb]，其他跟着pa后面的元素会通过图片2操作对路径进行压缩且更新d[i]的值

• 3、路径压缩的过程

  • 1、找到根结点root

  • 2、计算父节点到上一个父节点的距离d[p[x]],同时进行路径压缩(路径压缩后，上一父节点即root)

  • 3、d[x]初始值是x到p[x]的距离，更新d[x]值，即d[x] = d[x] + d[p[x]]
    -
    注意：题目C操作问的是间隔多少条船，若a != b，则间隔abs(d[a] - d[b]) - 1条船，若a == b，则间隔0条船

[以上思路为自己太懒为图省事摘抄其他大神的思路以便自己理解]：https://www.acwing.com/solution/content/11000/

[AC代码】

#include <iostream>
#include <cstring>
using namespace std;
const int N = 30010;
int m;
int p[N], sz[N], d[N];	//sz[]表示集合长度，d[i]表示i到p[i]的距离 

int find(int x)	//令p[x]更新为root，相应的d[x]也更新为x到root的距离
{
    if (p[x] != x)
    {
    //注意：这里的顺序不能乱，因为递归操作中加入了求d[x]，递归求路径压缩d[]，最后将根节点赋给p[x]
        int root = find(p[x]);
        d[x] += d[p[x]];
        p[x] = root;
    }
    return p[x];
}

int main() {
	int T;
	cin >> T;
	//初始化每个集合 
	for(int i = 1; i < N; i ++) {
		p[i] = i;
		d[i] = 0;
		sz[i] = 1;
	}
	
	while(T --) {
		char op[2];
		int a, b;
		scanf("%s%d%d", op, &a, &b);
		int pa = find(a), pb = find(b);
		
		if(op[0] == 'M') {
			if(pa != pb) {
				d[pa] = sz[pb];
				sz[pb] += sz[pa];
				p[pa] = pb;
			}
		} else {
			if(pa != pb)	puts("-1");
			else {	//可能a == b，故最小为0 
				printf("%d\n", max(0, abs(d[a] - d[b]) - 1));
			}
	
		}	
	}
	return 0;
}