POJ 1703 （多集合 并查集）

/*解决这题的关键在于，维护多种集合的关系；由于给的关系是两对立的，就需要第一个元素的反面是第二个元素的集合，第二个元素的反面是第一个元素的集合。有利于判断两个元素是同一集合，还是不能判断；比如，输入1 5 5 A(1,2);这时，1的反面是6 ，2的反面是7；他们的集合各不相同，所以不能判断；再输入D(1,2);由于我上面定的规矩，1 和 7 同一个集合，2 和 6 在一个集合，这个时候再输入，A(1,2);只要判断 1 ,和 2 的反面7  是不是在同一个集合就可以了，是，表示1，2是对立，在不同的集合。再输入 D(2, 4); 这时候，1 ，7， 4，  在一个集合， 2， 6， 9 在一个集合， 再输入 A(1 ,4)， 只要判断 1 ,和 4  是不是在同一个集合就可以了，是，表示1，4是在同一个的集合。就这样依次循环； */ #include <iostream>#include <stdio.h>using namespace std;int tree[100001 * 2];int drop[100001 * 2];int testCount,N,M;void init(){scanf("%d",&N);for(int i=1; i<= N*2; ++i){drop[i] = 0;tree[i] = i;}}int getRoot(int x){if(x == tree[x])return x;return tree[x] = getRoot(tree[x]);}void Union(int x,int y){x = getRoot(x);y = getRoot(y);if(x == y) return ;if(drop[x] < drop[y])  tree[x] = y;else{tree[y] = x;if(drop[x] == drop[y]) drop[x]++;}}bool same(int a, int b){return getRoot(a) == getRoot(b);}void show(){for(int i=1; i<=N; ++i){printf("%d =  %d %d\n",i, tree[i],drop[i]);}}int main() {int a,b;char ch;scanf("%d",&testCount);while(testCount--){init();scanf("%d",&M);while(M--){scanf("\n%c %d %d",&ch, &a,&b);if(ch == 'D'){Union(a,b+N);Union(a+N,b);//show();}else{if(same(a,b)) printf("In the same gang.\n");else if (same(a,b+N)) printf("In different gangs.\n");else printf("Not sure yet.\n");}}}}