[NOI2002] 银河英雄传说

题目链接：###

点我

题目分析：###

带权并查集的题，记一个\(d\)数组维护每个战舰前面有多少战舰，再在每个树根上记录一个集合大小\(sum\)，每次进行\(get_father\)的时候维护一下\(d\)数组，合并的时候更新集合大小。由于写了路径压缩，每次合并的时候还要把之前集合的\(sum\)更新到\(d\)数组上。

代码：###

#include<bits/stdc++.h>
#define N (1000000+5)
using namespace std;
inline int read(){
	int cnt=0,f=1;char c;
	c=getchar();
	while(!isdigit(c)){
		if(c=='-') f=-f;
		c=getchar();
	}
	while(isdigit(c)){
		cnt=cnt*10+c-'0';
		c=getchar();
	}
	return cnt*f;
}

int t,fa[N],d[N],sum[N],x,y;
char c;

int get_father(int x){
	if(fa[x]==x)return x;
	int root=get_father(fa[x]);
	d[x]+=d[fa[x]];
	return fa[x]=root;
}

void merge(int x,int y){
	int a=get_father(x);
	int b=get_father(y);
	fa[a]=b;d[a]=sum[b];
	sum[b]+=sum[a];
}

int query(int x,int y){
	int a=get_father(x);
	int b=get_father(y);
	if(a!=b) return -1;
	else {
	int ans=abs(d[x]-d[y])-1;
	return ans;
	}
}

int main(){
	t=read();
	for(register int i=1;i<=30001;i++) fa[i]=i,d[i]=0,sum[i]=1;
	for(register int i=1;i<=t;i++) {
		cin>>c;x=read();y=read(); 
		if(c=='M') merge(x,y);
		if(c=='C') printf("%d\n",query(x,y));
	}
	return 0;
}