银河英雄传说,本咸鱼刷的第一道NOI的题,估计是最简单的

来源:互联网 发布:javascript 防止csrf 编辑:程序博客网 时间:2024/05/06 14:09

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;} 

这就做完了,还挺简单的吗,就一个带权并查集就解决,所以说是最简单。


原创粉丝点击