[NOI2002]银河英雄传说————并查集+前缀和

最新推荐文章于 2023-11-22 12:58:13 发布

wly127

最新推荐文章于 2023-11-22 12:58:13 发布

阅读量252

点赞数

分类专栏：图论

本文链接：https://blog.csdn.net/wly1127/article/details/102961648

版权

图论专栏收录该内容

14 篇文章 0 订阅

订阅专栏

题解：本题主要考察并查集。
简要题意：有 $T$ 条指令,有两种操作：
$M$ 指令，将第 $i$ 号所在的队列，接至第 $j$ 号战舰所在的战舰队列的尾部
$C$ 指令，第 $i$ 号战舰与第 $j$ 号战舰当前是否在同一列中，如果在同一列中，那么它们之间布置有多少战舰。
1.并查集+前缀和：用并查集来合并区间，记录区间战舰数量。
求两个点的距离的问题，便想到用前缀和来实现： $f r o n t [i]$ 表示飞船 $i$ 到其所在队列队头的距离, $i$ 到 $j$ 的距离为 $a b s (f r o n t [i] - f r o n t [j]) - 1$
$n u m [i]$ 表示以 $i$ 为队头的队列的飞船数量
代码如下：

#include<bits/stdc++.h>
using namespace std;
int f[30001],front[30001],num[30001];//bcj,到头数,队列数
int n,T;
char s;
int find(int k)
{
    if(f[k]==k)return f[k];
    int fa=find(f[k]);
    front[k]+=front[f[k]];
    return f[k]=fa;
}
int main(){
    cin>>T;
    for(int i=1;i<=30000;i++)
    {
        f[i]=i;
        front[i]=0;num[i]=1;
    }
    for(int i=1;i<=T;i++)
    {
        cin>>s;
        int xx,yy;
        cin>>xx>>yy;
        int x=find(xx);
        int y=find(yy);
        if(s=='M')
        {
            front[x]+=num[y];
            f[x]=y;
            num[y]+=num[x];
            num[x]=0;
        }
        if(s=='C')
        {
            if(x!=y)cout<<"-1"<<endl;
            else cout<<abs(front[xx]-front[yy])-1<<endl;
        }
    }
    return 0;
}