POJ - 1182 食物链（并查集）

POJ - 1182 食物链

这道题目我们主要是要开三个拓展的域,也就是天敌域,同类域,以及捕食域.
1.如果x,y是同类,但是x的捕食域有y,那么错误
2.如果x,y是同类,但是x的天敌域有y,那么错误
3.如果x是y的天敌,但是x的同类域中有y,那么错误
4.如果x是y的天敌,但是x的天敌域中有y,那么错误

//这里我们将三个域,分别转化为了n,n+n,n+n+n.因为读入方面特别烦.
#include<stdio.h>
int fa[200000];
int n,m,k,x,y,ans;
int get(int x)
{
    if(x==fa[x]) 
        return x;
    return fa[x]=get(fa[x]);
}
void merge(int x,int y)
{
    fa[get(x)]=get(y);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=3*n;i++) 
        fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&k,&x,&y);
        if(x>n || y>n) 
            ans++;
        else if(k==1)
        {
            if(get(x)==get(y+n) || get(x)==get(y+n+n)) //如果x,y是同类,但是x是y的捕食中的动物,或者x是y天敌中的动物,那么错误.
                ans++;
            else
            {
                merge(x,y);
                merge(x+n,y+n);
                merge(x+n+n,y+n+n);
            }
        }
        else
        {
            if(x==y || get(x)==get(y) || get(x)==get(y+n)) //x就是y,或者他们是同类,再或者是y的同类中有x
                ans++;//都是假话
            else
            {
                merge(x,y+n+n);//y的捕食域加入x
                merge(x+n,y);//x的天敌域加入y
                merge(x+n+n,y+n);//x的捕食域是y的同类域.
            }
        }
    }
    printf("%d\n",ans);
}
//x是同类域.
//x+n是捕食域
//x+n+n是天敌域