POJ 1703 Find them, Catch them

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

题目大意：Tadu City里面有两个黑帮团伙Gang Dragon和Gang Snake，一共有n名团伙成员(还不知道属于这两个黑帮的哪一个)。现在警察局有一些信息，每条信息包含2个人编号，表示这2个人属于不同的帮派。问给你2个人的编号，能否确定他们是否属于同一个帮派。

题目分析：并查集。用opp[x]表示和x属于不同黑帮的一个成员编号，只需要知道一个就可以了，因为根据这一个就可以查询出其他的，属于同一黑帮的必定在同一颗树上。

AC代码：

/********************** * * * Author:surpasser * * Date:2010-9-23 * * Problem:Find them, * * Catch them * * * **********************/ #include<cstdio> #include<string> #include<cmath> #include<algorithm> using namespace std; const int MAXN = 500001; int p[MAXN]; int rank[MAXN]; int opp[MAXN]; //初始化 void Init_Set(int n) { for(int i=0;i<=n;i++) { p[i]=i; rank[i]=0;//按秩合并提高查找效率 opp[i]=0;//和i不是同一个组织的罪犯编号，初始化为0，表示还没出现 } } //查找并压缩路径 int Find_Set(int x) { if(p[x]==x) return x; return p[x]=Find_Set(p[x]); } //合并 void Union_Set(int x,int y) { x=Find_Set(x); y=Find_Set(y); if(x==y) return ; if(rank[x]==rank[y]) { p[y]=x; rank[x]++; } else { if(rank[x]>rank[y]) p[y]=x; else p[x]=y; } } int main() { //freopen("input.in","r",stdin); int ntest,n,m; char op; int a,b; scanf("%d",&ntest); while(ntest--) { scanf("%d %d/n",&n,&m); //getchar(); Init_Set(n); for(int i=1;i<=m;i++) { scanf("%c %d %d/n",&op,&a,&b); //getchar(); //printf("%c %d %d/n",op,a,b); if(op=='D') { if(opp[a]==0&&opp[b]==0)//第一种情况：两个罪犯都没出现过 { opp[a]=b; opp[b]=a; } else if(opp[a]==0)//第二种情况：a罪犯没出现过，b罪犯出现过 { opp[a]=b; Union_Set(a,opp[b]); } else if(opp[b]==0)//第三种情况：a罪犯出现过，b罪犯没出现过 { opp[b]=a; Union_Set(b,opp[a]); } else{//第四种情况：两个罪犯都出现过 Union_Set(a,opp[b]); Union_Set(b,opp[a]); } } else{ if(Find_Set(a)==Find_Set(b)) { printf("In the same gang./n"); } else if(Find_Set(a)==Find_Set(opp[b])) { printf("In different gangs./n"); } else printf("Not sure yet./n"); } } } return 0; }