poj 1703
并查集。
用一个opposite数组保存一个人a的对立人。对于输入的一个人b和c,只要判断b和opposite[c]是否是一组就好了。
也可以不用opposite数组,把并查集开到2倍的大小,用前n个保存数据,使用后n个作为opposite,一样的。
#include <iostream>
using namespace std;
const int MAX_NUM = 10000005;
int n, m, t;
int par[MAX_NUM], rank[MAX_NUM], opposite[MAX_NUM];
void init(int n) {
for(int i=0; i<n; i++) {
par[i] = i;
rank[i] = 0;
opposite[i] = 0;
}
}
int find(int x) {
if(par[x] == x)
return x;
else
return par[x] = find(par[x]);
}
void unit(int x, int y) {
x = find(x);
y = find(y);
if(x == y) return;
if(rank[x] < rank[y])
par[x] = y;
else {
par[y] = x;
if(rank[x] == rank[y]) rank[x]++;
}
}
bool same(int x, int y) {
return find(x) == find(y);
}
void solve(char c, int x, int y) {
if(c == 'D') {
if(opposite[x]==0 && opposite[y]==0) {
opposite[x] = y;
opposite[y] = x;
} else if(opposite[x] == 0) {
opposite[x] = y;
unit(x, opposite[y]);
} else if(opposite[y] == 0) {
opposite[y] = x;
unit(y, opposite[x]);
} else {
unit(x, opposite[y]);
unit(y, opposite[x]);
}
} else if(c == 'A') {
if(same(x, y))
printf("In the same gang.\n");
else if(same(x, opposite[y]))
printf("In different gangs.\n");
else
printf("Not sure yet.\n");
}
}
int main() {
freopen("in.txt", "r", stdin);
scanf("%d", &t);
while(t--) {
scanf("%d%d", &n, &m);
init(n * 2);//要开2*n的大小
char c[2];
int a, b;
while(m--) {
scanf("%s%d%d", &c, &a, &b);
solve(c[0], a, b);
}
}
fclose(stdin);
return 0;
}