poj1182

本文地址：https://www.cnblogs.com/maplefighting/p/9123814.html

题目名称：食物链

链接：http://poj.org/problem?id=1182

题意：中文题，自己看

思路：考察并查集。由于有三种动物，且不知种类，所以我们得分为三组i-A,i-B,i-C。i-X表示i属于种类X

若x和y同种，则合并x-A和y-A，x-B和y-B，x-C和y-C.

若x吃y，则合并x-A和y-B，x-B和y-C，x-C和y-A

代码如下：

 

#include<cstdio>
using namespace std;
const int MAX_K = 100005;
int n, k;
int f[50000 * 3 + 5];
void init(int n) {
    for(int i = 0; i < n; i++){
        f[i] = i;
    }
}
int found(int x) {
    return f[x] == x ? x : f[x] = found(f[x]);
}
void unite(int x,int y) {
    x = found(x), y = found(y);
    if(x != y){
        f[x] = y;
    }
    return ;
}
bool same(int x,int y) {
    return found(x) == found(y);
}
int main(){
    scanf("%d%d", &n, &k);
    init(n * 3);
    int sum = 0;
    while(k--) {
        int d, x, y;
        scanf("%d%d%d", &d, &x, &y);
        x--; y--;
        if(x < 0 || y < 0 || x >= n || y >= n) {
            sum++;
            continue;
        }
        if(d == 1) {
            //x和y为同类，判断x吃y或者y吃x为假
            if(same(x, y + n) || same(y, x + n)) {
                sum++;
            }
            else {
                unite(x, y);
                unite(x + n, y + n);
                unite(x + 2 * n, y + 2 * n);
            }
        }
        else {
            //x吃y，判断x和y同类，或者y吃x为假
            if(same(x, y) || same(y, x + n)){
                sum++;
            }
            else{
                unite(x, y+n);
                unite(x + n, y + 2 * n);
                unite(x + 2 * n, y);
            }
        }
    }
    printf("%d\n",sum);
    return 0;
}