codevs 1051[NOI 2002]银河英雄传说

AC通道:http://codevs.cn/problem/1540/

[分析]

因为最近在做并查集的专题，所以很容易就可以发现这道题可以用并查集解决。

我们可以把战场的每一列看作一个集合，初始时每个集合中只有一个元素。

M指令可以看作合并集合，这令我们想到并查集。

定义fa数组，dis数组和num数组，分别表示并查集中每个元素的父亲，距自己的父亲的距离和该集合中的元素个数。

只需要合并和路径压缩的时候注意一下就可以了。

C指令可以利用类似前缀和的思想。

#include <iostream>
#include <cstdio>

using namespace std;

const int N=30001;
int T;
int fa[N],dis[N],num[N];

inline int abs(int x){
	return (x>0)?x:(-x);
}

void dfs(int end,int now){
	if(now!=end)dfs(end,fa[now]);
	dis[now]+=dis[fa[now]];
	fa[now]=end;
}

int find(int x){
	int tmp=x,pre;
	while(tmp!=fa[tmp])tmp=fa[tmp];
	pre=tmp;
	if(pre!=x)dfs(pre,x);
	return tmp;
}

void merge(int l,int r){
	int fl=find(l),fr=find(r);
	fa[fl]=fr;dis[fl]=num[fr];
	num[fr]+=num[fl];
}

int main(){
	scanf("%d",&T);
	for(int i=1;i<=30000;i++){fa[i]=i;num[i]=1;}
	for(int i=1;i<=T;i++){
//		printf("%d:\n",i);
		char c[10];
		int l,r;
		scanf("%s%d%d",c,&l,&r);
		if(c[0]=='M'){
			merge(l,r);
		}
		else{
			if(find(l)!=find(r))printf("-1\n");
			else printf("%d\n",abs(dis[l]-dis[r])-1);
		}
	}
	return 0;
}