P1196 [NOI2002] 银河英雄传说 题解

题目传送门

分析

可以看到，题目中涉及到很多合并和查询操作，所以我们可以用并查集来维护每列的信息。

不过这题需要而外维护两个数组 $di{s}_i$ 和 $si{z}_i$，分别表示距离序列头部的距离和以 $i$ 做头的序列长度。

合并时需要更新 $siz$ 和 $dis$。

查询 $i$ 和 $j$ 之间的飞船个数就是 $|di{s}_i-di{s}_j|-1$。

代码

#include <bits/stdc++.h>
using namespace std;
int t;
int fa[30005];
int dis[30005];
int siz[30005];
int find(int x){
	if(fa[x]==x)return x;
	int f=find(fa[x]);
	dis[x]+=dis[fa[x]];
	return fa[x]=f;
}
void merge(int i,int j){
	int x=find(i),y=find(j);
	if(x!=y){
		dis[x]+=siz[y];
		fa[x]=y;
		siz[y]+=siz[x];
		siz[x]=0;
	}
}

int main(){
	cin>>t;
	for(int i=1;i<=30000;i++)fa[i]=i,dis[i]=0,siz[i]=1;
	while(t--){
		char ch;
		int i,j;
		cin>>ch>>i>>j;
		if(ch=='M'){
			merge(i,j);
		}else{
			if(find(i)!=find(j))cout<<-1<<endl;
			else cout<<abs(dis[i]-dis[j])-1<<endl;
		}
	}
	system("pause");
	return 0;
}