银河英雄传说/清北day1 t3 cubes

银河英雄传说/清北day1 t3 cubes

–加权并查集

---

题目分析

题意简化：

现在有n个积木，维护两种操作：

1. 将这一列的积木移动到另一列积木的上方
2. 查询两块积木的距离（若两块积木不在同一列则输出-1）

显然需要并查集维护两块积木是否在同一列中。

那么如何维护同一列的积木的距离？

首先我们需要一个head[i]数组维护该点到其所在队头的距离，还需要一个len[i]来维护以i为队头的队列的长度。关于fa[i]，，，，我们都知道的。

怎么更新head[i]和len[i]呢?

显然当我们把一个队列a与另一个队列b合并时新的队列c的len_c=len_a+len_b
所以在merge中我们使len[b]+=len[a]，因为合并后b在a的前面，所以不再存在a队列
（len[x]=0）了，而且我们顺便在find中更新了b中的每一个head[i]

PS：为什么是head[i]+=head[fa[i]]?
因为我们可能直接合并的不止是一个积木，，而是一列积木，因为这一列积木a是放在b的后面（上面）的，所以说，是head[i]+=head[fa[i]];

代码

#include<bits/stdc++.h>
using namespace std;
int n;
int fa[30010],len[30010],head[30010];
int find(int now)
{
    if(now!=fa[now]){
        int father=fa[now];
        fa[now]=find(fa[now]);
        head[now]+=head[father];
    }
    return fa[now];
}
void merge(int x,int y)
{
	x=find(x),y=find(y);
	fa[x]=y;
	head[x]+=len[y];
	len[y]+=len[x];
	len[x]=0;
}
int main()
{
	ios::sync_with_stdio(false);cin.tie(0);
	for(int i=1;i<=30000;i++)fa[i]=i,len[i]=1;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		char c;
		cin>>c;
		if(c=='M')
		{
			int a,b;
			cin>>a>>b;
			merge(a,b);
		}
		if(c=='C')
		{
			int a,b;
			cin>>a>>b;
			int fx=find(a),fy=find(b);
			if(fx!=fy)cout<<"-1"<<endl;
			else cout<<abs(head[a]-head[b])-1<<endl;
		}
	}
	return 0;
}