poj1182 并查集+关系维护

poj1182

题意：有 N 只动物， 编号为1 - N，这 N 只动物为别为 A、B、C 三类中的一类，已知：A 吃 B、B 吃 C、C 吃 A，同类不互吃。给你 K 条条件（a b c）： a == 1 b 和 c 为同类； a == 2 为b 吃 c。条件从上到下依次满足（当编号大于N，也为错），问你有几条错误条件。

思路：并查集。。。一般并查集我们只需维护一种关系，这道题有两种关系同类和捕食关系。如何维护：因为我们并不知道该动物究竟属于哪类，那么我设 1 ~ N 为 A 类， N+1 ~ N+N 为 B 类，1+2N ~ 3N 为 C 类。其实 1、1+N、1+2N 是同个动物，这样我们就不用纠结于他是A类会怎样、B类会怎样、C类会怎样了。

1、若条件为 1 b c 时，说明 b 和 c 为同类，那么我先查看之前是否有条件说 b 和 c 为捕食关系，若有错误条件数++，否则合并（b， c）、（b+N， c+N）、（b+2N，c+2N）；
2、若条件为 2 b c 时，说明 b 和 c 为捕食关系，则查看之前是否有条件说b 和 c 为同类，若没有则将合并（b， c+N）、（b+N， c+2N）、（b+2N，c）。

注：只能单组数据。。。while（~scanf。。。）什么的会 WA 的

代码实现：

#include <iostream>
#include <cstdio>

using namespace std;

int num[160000];
int find_fa(int x)
{
    while(num[x] != x)
    {
        x = num[x];
    }

    return x;
}

bool Same(int a, int b)
{
    int aa = find_fa(a);
    int bb = find_fa(b);

    if(aa == bb)
        return true;

    return false;
}

void Join(int a, int b)
{
    int aa = find_fa(a);
    int bb = find_fa(b);

    if(aa != bb)
    {
        num[aa] = bb;
    }


    return ;
}

int main()
{
    int N, K;
    scanf("%d%d", &N, &K);
    int a, b, c;
    int ans = 0;
    memset(num, -1, sizeof(num));
    for(int i = 0; i < K; i++)
    {
        scanf("%d%d%d", &a, &b, &c);

        if(b > N || c > N)
        {
            ans++;
            continue;
        }

        if(num[b] == -1)
        {
            num[b] = b;
            num[b+N] = b + N;
            num[b+2*N] = b +  2*N;
        }  
        if(num[c] == -1)
        {
            num[c] = c;
            num[c+N] = c + N;
            num[c+2*N] = c + 2*N;
        } 

        if(a == 1)
        {
            if(Same(b, c+N) || Same(c, b+N))
                ans++;
            else
            {
                Join(b, c);
                Join(b+N, c+N);
                Join(b+2*N, c+2*N);
            }
        }
        else
        {
            if(Same(b, c) || Same(c, b+N))
                ans++;
            else
            {
                Join(b, c+N);
                Join(b+N, c+2*N);
                Join(b+2*N, c);
            }
        }
    }

    printf("%d\n", ans);

    return 0;
}

若有错，请大家多多指教^ ^