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;
}