POJ 1182 食物链（分层并查集）

题目链接：
POJ 1182 食物链
题意：
有n个动物，编号1–n，每个动物属于A,B,C三种中的一种，并且A种动物吃B种动物，B种动物吃C种动物，C种动物吃A种动物。
有K条语句。格式是：d，x，y。
d=1时表示x和y是属于同一种类，
d=2时表示x吃y。
这条语句如果满足下列条件之一就是错误的语句：
1） 当前的话与前面的某些真的话冲突，就是假话；
2） 当前的话中X或Y比N大，就是假话；
3） 当前的话表示X吃X，就是假话。
问k条语句中一共有多少条语句是错误的？
分析：
在数组pre[3*maxn]中，1–n表示种类A，n+1–2*n表示种类B，2*n+1–3*n表示种类C。
当d=1时：先检查x和y是否属于不同的种类，如果是不同种类，则ans++，否则就将x，y同属于A,B,C依次mix；
当d=2时：先检查x和y是否属于同一种类和y吃x的情况，如果是的，则ans++，否则就将x属于A，y属于B；x属于B，y属于C；x属于C，y属于A三种情况mix.
测试数据只有一组。

#include <iostream>#include <cstdio>#include <cmath>#include <cstring>#include <algorithm>using namespace std;const int maxn = 50010;int n, k, d, x, y;int pre[3 * maxn];int find(int x){    return pre[x] == x ? x : pre[x] = find(pre[x]);}void mix(int x, int y){    int fx = find(x);    int fy = find(y);    if (fx != fy)    {        pre[fx] = fy;    }}int main(){#ifdef LOCAL    freopen("in.txt", "r", stdin);    //freopen("out.txt", "w", stdout);#endif    scanf("%d%d", &n, &k);    for (int i = 1; i <= 3 * n; i++)        pre[i] = i;    int ans = 0;    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;        }        if (d == 1)        {            if (find(x) == find(y + n) || find(y) == find(x + n)) ans++;            //因为当x和y属于不同种类时无非是x吃y或者y吃x，而且每种捕食关系的三种种类归属情况同时入队列，            //所以检查x和y属于不同种类其实就是检查 x和y+n 与 x+n和y 是否属于同一集合。（前者是x吃y，后者是y吃x）            else            {                mix(x, y);//同属A                mix(x + n, y + n);//同属B                mix(x + 2 * n, y + 2 * n);//同属B            }        }        else        {            if (find(x) == find(y) || find(y) == find(x + n)) ans++;            //前者是检查x和y是否属于同一种类，后者是检查是否是y吃x            else            {                mix(x, y + n);//x属于A,y属于B                mix(x + n, y + 2 * n);//x属于B,y属于C                mix(x + 2 * n, y);//x属于C，y属于A            }        }    }    printf("%d\n", ans);    return 0;}