A Bug's Life POJ - 2492种类并查集

题目→A Bug's Life

感觉种类并查集挺有意思的，做个思路记录~~

//种类并查集
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int parent[2010];
int rel[2010];//注意：rel为x与根的关系！！！！ relation
/*
若两虫子同性 ，则rel值为0
若虫子异性 ，则rel值为1
 */
int find(int x){
    if(x==parent[x])
        return x;
    int t=parent[x];
    parent[x]=find(t);

    rel[x]=(rel[x]+rel[t])%2;//可枚举情况得出式子

    return parent[x];
}
void merge(int x,int y){
    int rx = find(x), ry = find(y);
    if (rx != ry)
    {
        parent[rx] = ry;
        rel[rx] = (rel[x] + 1 + rel[y]) % 2;//一定程度上可与向量相联系，可枚举情况得出式子
    }           //其中的1是x与y的关系
}
int main(){
    int t;
    scanf("%d",&t);
    for(int z=1;z<=t;z++){
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;i++){
            parent[i]=i;
        }
        memset(rel,0,sizeof(rel));
        bool flag=0;
        int x,y;
        for(int i=0;i<m;i++){
            scanf("%d%d", &x, &y);
            if(flag) 
                continue; 
            int rx=find(x);
            int ry=find(y);
            /*在输入过程中，对于每一次输出，判断树中是否存在同性恋
             若两只虫子原本已在parent树中，则根节点肯定会相同，并且此时两虫子与根的关系也有了
             所以做判断是否出现同性恋（与根的关系都为0或都为1，说明是同性恋，否则异性恋~）
             */
            if(rx!=ry){//说明至少有一只虫子之前还未曾加入到树中 所以肯定不能判断出该虫子与已在树中的任何一虫子为同性恋，所以直接合并到树中
                merge(x,y);
            }else{
                if(rel[x]==rel[y])
                    flag=1;
            }
        }
        if(flag){
            printf("Scenario #%d:\nSuspicious bugs found!\n", z);
        }else{
            printf("Scenario #%d:\nNo suspicious bugs found!\n", z);
        }
        if(z!=t) putchar('\n');
    }

    return 0;
}