POJ 1703 Find them, Catch them 并查集

原题见POJ 1703

N个人，M种输入，要么是声明两者不是同伙，要么是询问两者是否是同伙。

思路

年幼不懂事，看着紫书上的秩做的，今天重新思考一番，其实原理和积木移动的想法如出一辙。

把人当作积木，可以相互堆叠，a[i]为积木i底下的积木的个数（不严格，也可以是和这个数同奇偶的数字，在每次get累加路径上所有的a才能得到），p[i]为这堆积木最低端的积木编号。初始a[i]=0,p[i]=i.

声明操作：将i所在的积木堆整个移动到j所在的积木堆上方。同时改变a[p[i]]的值，使得get操作之后a[i]和a[j]不同奇偶（不是同伙）。再将p[p[i]]指向p[j]。

询问操作：get操作之后p[i]==p[j]时，i和j才具有可比性（比较的传递性），a[i]和a[j]同奇偶才是同伙，否则不是。

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 100100
int n, m, p
, a
;
int get(int x){
if(x != p[x]){
int t = get(p[x]);
a[x] += a[p[x]]%2;
p[x] = t;
}
return p[x];
}
int main()
{
int T;
scanf("%d", &T);
while(T--){
scanf("%d%d", &n, &m);
char com[10];
int u, v;
for(int i = 0;i < n;i++){
p[i] = i;
a[i] = 0;
}
for(int i = 0;i < m;i++){
scanf("%s%d%d", com, &u, &v);
u--; v--;
get(u); get(v); //先get一下，得到u和v的团长
if(com[0] == 'A'){
if(p[u] != p[v]) printf("Not sure yet.\n");
else if(a[u] % 2 == a[v] % 2) printf("In the same gang.\n");
else printf("In different gangs.\n");
}
else{
a[p[v]] %= 2;
a[p[u]] = a[u]%2 == a[v]%2 ? 1:0; //合并的关键
p[p[u]] = p[v];
}
}

}
return 0;
}