AcWing240. 食物链

题目链接：240. 食物链 - AcWing题库

输入样例：

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例：

3

思路1：带拓展域的并查集，规定fa数组的1~n表示该动物是A类动物，n + 1 ~ 2 * n表示该动物是B类动物，2 * n + 1 ~ 3 * n表示该动物是C类动物。

当d为1的时候，应当先检查两者是否有捕食和被捕食的关系，若没有才将两者合并。

当d为2的时候，应当先检查两者是否是同类，或者b是不是a的捕食者。

代码1：

#include<bits/stdc++.h>
using namespace std;
const int N = 5e4 + 5;
int n, k, ans, fa[4 * N];

int find(int x){
    if(x == fa[x])  return fa[x];
    return fa[x] = find(fa[x]);
}

void merge(int x, int y){
    int fx = find(x);
    int fy = find(y);
    if(fx != fy)  fa[fx] = fy;
}

void init(){
    for(int i = 1; i <= 4 * n; i++)  fa[i] = i;
}

int main(){
    cin >> n >> k;
    init();
    for(int i = 1; i <= k; i++){
        int d, x, y;  cin >> d >> x >> y;
        if(x > n || y > n || (d == 2 && x == y))  ans++;
        else if(d == 1){
            if(find(y + n) == find(x) || find(x + n) == find(y))   ans++;
            else{  //合并为同一类动物
                merge(x, y);
                merge(x + n, y + n);
                merge(x + 2 * n, y + 2 * n);
            }
        }
        else if(d == 2){
            if(find(x) == find(y) || find(x + n) == find(y))  ans++;
            else{  //设立捕食关系
                merge(x, y + n);
                merge(x + n, y + 2 * n);
                merge(x + 2 * n, y);
            }
        }
    }
    printf("%d\n", ans);
} 

思路2：带边权的并查集，非常神奇的一种做法，A、B、C三个之间的循环可以简化成0、1、2之间的循环。路径压缩的时候可以直接将子节点到父节点的距离将它们之间的关系算出来。

0表示子节点和父节点是同类

1表示子节点吃父节点

2表示子节点被父节点吃

在路径压缩的时候分成两种情况，假设z节点和y节点的关系已知：

1. z和y拥有共同的祖先x，d[z] = d[z] + d[y]

2. z和y不拥有共同的祖先，d[z]不变，d[x]（y的父节点）就变成-d[y] + y和z之间的关系 + d[z],之所以是-d[y]，是因为箭头方向反过来，关系也会转变，随着值也会改变。

代码2：

#include<bits/stdc++.h>
using namespace std;
const int N = 5e4 + 5;
int fa[N], n, k, d[N], ans;

int find(int x){
    if(x != fa[x]){
        int t = find(fa[x]);
        (d[x] += d[fa[x]]) %= 3;
        fa[x] = t;
    }
    return fa[x];
}

void init(){
    for(int i = 1; i <= n; i++) fa[i] = i;
}

int main(){
    cin >> n >> k;
    init();
    for(int i = 1; i <= k; i++){
        int r, x, y;  cin >> r >> x >> y;
        int fx = find(x), fy = find(y);
//        printf("******%d %d\n", fx, fy);
        if(x > n || y > n)  ans++;
        else if(r == 1){
            if(fx == fy){
                if(d[x] != d[y])  ans++;
            }
            else{  //合并同类
                fa[fx] = fy;
                d[fx] = (-d[x] + 0 + d[y] + 3) % 3;
            }
        }
        else{
            if(x == y)  ans++;
            else{
                if(fx == fy){
                    if((d[x] - d[y] + 3) % 3 != 1)  ans++;
//                    printf("^.^%d\n", (d[x] - d[y] + 3) % 3);
                }
                else{
                    fa[fx] = fy;
                    d[fx] = (1 + d[y] - d[x] + 3) % 3;
                }
            }
        }
//        printf("***%d %d\n", i, ans);
    }
    printf("%d\n", ans);
}