银河英雄传说/清北day1 t3 cubes
–加权并查集
题目分析
题意简化:
现在有n个积木,维护两种操作:
- 将这一列的积木移动到另一列积木的上方
- 查询两块积木的距离(若两块积木不在同一列则输出-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;
}