https://www.luogu.org/problem/show?pid=1196 链接传送
我们先进行题目篡改。
若把输出队尾到队首的距离改为在同队列里输出1,相信大家都会想到用并查集解决。但这道题要求到队首距离,那么我们可不可以创个数组记录某战舰到队首的距离,边进行并查集边更新该数组,查询时用后面的距离减去前面的距离。但怎么更新呢,似乎只用一个数组来实现有点麻烦,那我们再加一个数组记录队伍长度,这样子就可以这样解决:
代码如下:
#include<cstdio>
#include<string>
#include<iostream>
using namespace std;
int size[30001],dis[30001],fa[30001];
int find(int x)
{
if(fa[x]==x)return x;
int f=fa[x],o=find(fa[x]);
fa[x]=o;
dis[x]+=dis[f];//更新合并后到队首的距离
return o;
}
int merge(int u,int v)//合并函数
{
//寻找两个合并的舰队
int f1=find(u);
int f2=find(v);
if(f1!=f2)
{
fa[f2]=f1;
dis[f2]=size[f1];//因为以f2为首的舰队跟在f1舰队的后面,所以f2队首距离为该队列长度
size[f1]+=size[f2];//长度相加
}
}
int main()
{
string a;
int n;
scanf("%d",&n);
for(int i=1;i<=30000;i++)
{
fa[i]=i;
dis[i]=0;
size[i]=1;
}
for(int i=1;i<=n;i++)
{
cin>>a;
//分情况
if(a[0]=='M')
{
int u,v;
scanf("%d%d",&u,&v);
merge(u,v);
}
if(a[0]=='C')
{
int i,j;
scanf("%d%d",&i,&j);
int f1=find(i),f2=find(j);
if(f1!=f2)printf("-1\n");
else
{
int t;//t为两战舰之间的战舰数
if(dis[i]>=dis[j])
t=dis[i]-dis[j]-1;
else t=dis[j]-dis[i]-1;
printf("%d\n",t);
}
}
}
return 0;
}
这就做完了,还挺简单的吗,就一个带权并查集就解决,所以说是最简单。