问题描述
蒜头君在玩一种接龙的游戏,蒜头君有 30000 张卡片分别放在 30000 列,每列依次编号为 1,2,…,300001,2,…,30000。同时,蒜头君也把每张卡片依次编号为 1,2,…,30000
游戏开始,蒜头君让让第 ii 张卡片处于第i(i=1,2,…,30000) 列。然后蒜头君会发出多次指令,每次调动指令 M i j 会将第 i 张卡片所在的队列的所有卡片,作为一个整体(头在前尾在后)接至第 j 张卡片所在的队列的尾部。
蒜头君还会查看当前的情况,发出 CC ii jj 的指令,即询问电脑,第 ii 张卡片与第 jj 张卡片当前是否在同一个队列中,如果在同一列中,那么它们之间一共有多少张卡片。
聪明的你能不能编写程序处理蒜头君的指令,以及回答蒜头君的询问呢?
输入格式
第一行有一个整数 TT(5000001≤T≤500000),表示总共有 T 条指令。
以下有 T行,每行有一条指令。指令有两种格式:
M i j:i 和 j 是两个整数(300001≤i,j≤30000),表示指令涉及的卡片编号。你需要让第 i 张卡片所在的队列的所有卡片,作为一个整体(头在前尾在后)接至第 jj 张卡片所在的队列的尾部,输入保证第 i 号卡片与第 j 号卡片不在同一列。
C i j:i 和 j 是两个整数(1≤i,j≤30000),表示指令涉及的卡片编号。该指令是蒜头君的询问指令。
输出格式
如果是蒜头君调动指令,则表示卡片排列发生了变化,你的程序要注意到这一点,但是不要输出任何信息;
如果是蒜头君的询问指令,你的程序要输出一行,仅包含一个整数,表示在同一列上,第 ii 号卡片与第 jj 号卡片之间的卡片数目(不包括第i张卡片和第 j 张卡片)。如果第i 号卡片与第 j 号卡片当前不在同一个队列种中,则输出 -1。
样例输入
4
M 2 3
C 1 2
M 2 4
C 4 2
样例输出
-1
1
#include <bits/stdc++.h>
using namespace std;
const int maxn=5e6+2;
int pre[maxn];
int val[maxn];//存储到根节点距离的数组
int Size[maxn];//存储在该节点下集合的大小
void build(int n)
{
for(int i=0;i<=n;i++)
pre[i]=i,val[i]=0,Size[i]=1;
}
int find(int x)
{
if(pre[x]==x) return x;
int y=pre[x];
pre[x]=find(y);
val[x]+=val[y];//x到根节点的距离就等本身加上双亲到根节点的距离
return pre[x];
}
bool join(int x,int y)
{
int p=find(x);
int q=find(y);
if(p!=q){
pre[p]=q;
//val[q]+=val[p];
val[p]=Size[q];//p到根节点的位置,就等于q集合的大小
Size[q]+=Size[p];//q集合的大小增加了,p集合加入了进来
return true;
}
return false;
}
int main()
{
int n;scanf("%d",&n);
build(n);
while(n--){
char c;
int x,y;
cin>>c>>x>>y;
if(c=='M'){
join(x,y);
}
else{
if(find(x)!=find(y)) cout<<"-1"<<endl;
else cout<<abs(val[x]-val[y])-1<<endl;
}
}
return 0;
}