1196 银河英雄传说
练习并查集,并查集其实就是给二分图里铺垫,二分图和并查集都非常重要
我们得把题目进行提炼
有三万个战舰,初始化每一个战舰单独在一列,现在有n个指令
1.将x战舰所在作为整体,接到y战舰所在的列尾部
2.询问x与y是否在同一列,如果在,输出他们两个之间隔了多少战舰
其实很明显就是并查集了,不过感觉第二个操作比较难
我们可以简单的实现合并和查询是否联通,但是如何维护之间隔了多少战舰
可以利用类似于前缀和的数组sum,sum[x]表示x到fa[x]中间有多少战舰
然后就可以快乐的AC了
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int SIZE=3e4+5;
int fa[SIZE],sum[SIZE];//此点到队首的长度
int he[SIZE];//本列总长
int n;
char opt;
int u,v;
inline int findd(int x)
{
if(fa[x]==x)
return x;
int t1=fa[x],t2=findd(fa[x]);
fa[x]=t2;//路径压缩
sum[x]+=sum[t1];//计算长度
return t2;
}
int abs(int x)
{
return x<0?-x:x;
}
void un(int u,int v)
{
int t1=findd(u),t2=findd(v);
fa[t2]=t1;
sum[t2]=he[t1];//是节点到父亲的距离
he[t1]+=he[t2];//放了he[f2]个战舰
he[t2]=0;
}
int main()
{
for(int i=1;i<=30000;i++)
{
fa[i]=i;
he[i]=1;
}
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>opt;
cin>>u>>v;
if(opt=='M')//合并
un(v,u);
else
{
if(findd(u)!=findd(v))
cout<<-1<<endl;
else
cout<<abs(sum[u]-sum[v])-1<<endl;
}
}
return 0;
}