poj 1182 食物链 （带关系的并查集）

/*
 * 对每只动物建立3个元素A,B,C,位置分别为i, i+n, i+2*n
 * i-X 表示i属于种类X
 * 并查集中每一组表示组内所有元素代表的情况都同时发生或不发生
 */
#include<cstdio>
#include<cstring>
#include<iostream>
#include<string>
#include<algorithm>
using namespace std;
const int maxn=50000*3+5;
const int maxk=100000+5;
int p[maxn];

void make_set()
{
    memset(p, -1, sizeof(p));
}

int find_set(int x)
{
    return p[x]==-1 ? x : p[x]=find_set(p[x]);
}

void union_set(int x, int y)
{
    int fx=find_set(x), fy=find_set(y);
    if(fx==fy) return;
    p[fx]=fy;
}

bool same(int x, int y)
{
    return find_set(x)==find_set(y);
}


int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    int d, x, y;
    int ans=0;
    make_set();
    for(int i=0;i<k;i++)
    {
        scanf("%d%d%d", &d, &x, &y);
        if(x>n || y>n || (d==2 && x==y)) { ans++; continue; }
        x--; y--;

        if(d==1)//x y 属于同一类
        {
            if(same(x, y+n) || same(x, y+2*n))
                ans++;
            else
            {
                union_set(x, y);
                union_set(x+n, y+n);
                union_set(x+2*n, y+2*n);
            }
        }
        else if(d==2) {//x 吃 y
            if(same(x, y) || same(x, y+2*n))
                ans++;
            else
            {
                union_set(x, y+n);
                union_set(x+n, y+2*n);
                union_set(x+2*n, y);
            }
        }
    }
    printf("%d\n", ans);

    return 0;
}