Find them, Catch them

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
#define maxn 100010//一个帮派的上限
int set[maxn+maxn];//龙帮帮派,蛇帮帮派为k+n
int n,m;
int find(int d)//带路径压缩的并查集查找集合代表元素
{
    if(set[d]==0)//若d为集合代表元素,则返回
        return d;
    return set[d]=find(set[d]);//递归计算d所在集合的代表元素
}
int main()
{
    int t,a,b;
    char c;
    scanf("%d",&t);
    while(t--)
    {
        memset(set,0,sizeof(set));//清空
        scanf("%d%d",&n,&m);//输入测试实例
        for(int i=0;i<m;i++)
        {
            getchar();
            scanf("%c%d%d",&c,&a,&b);
            if(c=='A')//确定是否是属于同一帮派
            {
                if(find(a)!=find(b)&&find(a)!=find(b+n))
                    //若a和b不属于同一帮派,且a所属的帮派与b所属另一帮派
                    printf("Not sure yet.\n");
                else if(find(a)==find(b))//若属于同一帮派,则输出In the same gang.
                    printf("In the same gang.\n");
                else
                    printf("In different gangs.\n");
            }
            else
            {
                if(find(a)!=find(b+n))//若a所属的帮派不为b的另一帮派,则a设为b的另一帮派,b帮派设为a的另一帮派
                {
                    set[find(a)]=find(b+n);
                    set[find(b)]=find(a+n);
                }
            }
        }
    }
    return 0;
}

你可能感兴趣的:(Find them, Catch them)