pku1703 Find them, Catch them

题目链接：http://acm.pku.edu.cn/JudgeOnline/problem?id=1703

题意简述：两种操作，A，x，y和D，x，y。分别是问x和y是否属于同一类，和指示x和y是不同类。根据题目的输出来输出相应的返回信息。

解题思路：并查集+乱搞。反正我是想着想着就乱搞出来了，在一般的并查集当中加了一个num[i]数组表示与 i 相对的号码。然后就是简单的并查集操作了。

代码：

#include<stdio.h> using namespace std; const int Max=100005; int num[Max],root[Max],rank[Max]; int find(int a) { if(a!=root[a]) return (root[a]=find(root[a])); else return root[a]; } void merge(int a,int b) { int roota=find(a); int rootb=find(b); root[rootb]=roota; } int main() { int t; scanf("%d",&t); while(t--) { int n,m; scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { num[i]=0; root[i]=i; rank[i]=0; } bool flag=false; getchar(); for(int i=1;i<=m;i++) { char c; int a,b; scanf("%c%d%d",&c,&a,&b); getchar(); int roota=find(a),rootb=find(b); if(c=='D') { if(num[roota]) { if(rank[find(num[roota])]>=rank[rootb]) { if(rank[find(num[roota])]==rank[rootb]) rank[find(num[roota])]++; merge(num[roota],rootb); } else merge(rootb,num[roota]); } if(num[rootb]) { if(rank[find(num[rootb])]>=rank[roota]) { if(rank[find(num[rootb])]==rank[roota]) rank[find(num[rootb])]++; merge(num[rootb],roota); } else merge(roota,num[rootb]); } if(!num[rootb]) num[rootb]=roota; if(!num[roota]) num[roota]=rootb; continue; } if(c=='A') { if(roota==rootb) printf("In the same gang./n"); else if(find(num[roota])==rootb) printf("In different gangs./n"); else printf("Not sure yet./n"); } } } return 0; }