poj1703

链接：点击打开链接

题意：有N个罪犯，M条关系有两种可能，分别为"D[a][b]"代表a，b为不同团体，"A [a] [b]"则输出相应的关系

代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int par[1000005],rank[1000005];
int T,N,M;
void init(int n){
    int i;
    for(i=0;i<=2*n;i++){
        par[i]=i;
        rank[i]=0;
    }
}
int find(int x){
    if(par[x]==x)
    return par[x];
    return par[x]=find(par[x]);
}
void unite(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);
}                                   //并查集模板
int main(){
    int i,j,x,y;
    char c;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&N,&M);
        init(2*N);
        for(i=0;i<M;i++){
            scanf(" %c",&c);
            scanf("%d%d",&x,&y);
            if(c=='D'){             //用x表示属于x属于第一个团体，x+N属于第二个团体，用两个人的关系
                unite(x,y+N);       //维护每个集合，每个集合的意义在于集合中所有的事情一定同时发生
                unite(y,x+N);       //也就是所有人都符合逻辑关系，而不一定是所有人都是同一个团体
            }
            else{
                if(same(x,y))
                puts("In the same gang.");
                else if(same(x,y+N)||same(x+N,y))
                puts("In different gangs.");
                else
                puts("Not sure yet.");
            }
        }
    }
    return 0;
}