POJ 1703 Find them, Catch them 种类并查集

输入n，m表示一共有n个节点 ，m个行条件+询问
A表示询问A与B是不是同类
D表示 A与B不是同类
我们用两个节点rank值的和值来表示是否为同类
和值%2==0表示为同类

和值%2==1表示为不同类

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <limits.h>
#include <math.h>
#include <iostream>
#include <algorithm>
using namespace std;
#define N 500000+5
int fa[N],rank[N];
void init(int n){//初始化
    for(int i=0; i<=n; i++){
        fa[i]=i;
        rank[i]=0;
    }
}
int find(int x)//找父节点
{
    if(x==fa[x])    return x;
    else
    {
        int f=fa[x];
        fa[x]=find(fa[x]);//递归
        rank[x]=(rank[x]+rank[f])%2;//更新rank值
        return fa[x];
    }
}
void merge(int x,int y)
{
    int fx=find(x),fy=find(y);
    if(fx==fy)  return ;//
    else
    {
        fa[fy]=fx;  //更新fa
        rank[fy]=(rank[x]-rank[y]+2+1)%2;
    }
}
int main()
{
 //   freopen("in.txt","r",stdin);
    int n,m,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        init(n);
        int ans=0;
        for(int i=0; i<m; i++)
        {
            int b,c;
            char a;
            getchar();
            scanf("%c%d%d",&a,&b,&c);
            if(a=='D')
                merge(b,c);
            else
            {
                int x=find(b),y=find(c);
                if(y!=x)//没有相同的父节点表示先前没有联系
                    printf("Not sure yet.\n");
                else if(rank[b]==rank[c]&&x==y)//父节点相同 并且rank值%2==0（0 +0或者1+1）
                    printf("In the same gang.\n");
                else printf("In different gangs.\n");
            }
        }
    }
    return 0;
}