NOI食物链 - 拓展域并查集/带权并查集

合并拓展域的时候要考虑周全，因为信息其实是残缺的，我们得自己补上残缺的信息
比如说x吃y，隐含z吃x，也就是说y的捕食和x的天敌在一个集合里，以及，x的捕食和y一个集合，y的天敌和x一个集合
一般3种关系要合并3次集合
下面是带权并查集的做法
一开始可能有点难理解，不过慢慢来
设r[i]为i到其根的关系 0 为同类 1 为捕食 2 为被捕食
想象一下，如果i到根有好多个1，那么把这些1加起来%3之后就是i和根的关系
有一些0，1，2都是一样的
关键是有一个方向的问题
如果说i - > j的关系为x
那么j - > i的关系为（-x+3）%3 为什么要加3%3 这是个规律，举几个例子试试就知道了，可以说+3%3是c++处理负数%一个数的方式，c++给出负的余数，但是可以移动到正数
加减取决于“方向”
而合并时有一个等式 x fx y fy 假设fx到fy的有向边 那么先求出x和fy的关系，设fy为所有点的根，但此时rx还没有改变， 因为xy是给定好关系的 xy + ry = (x->fy = rx) = rx+fxy
则fxy = xy+ry-rx（rfx为x以fx为根时的r值，因为没有改变）
那么
如果给出的两个点a，b已在同一棵树上，那么他们的关系是(ra - rb)%3

#include <algorithm>
#include <iostream>
#include <cstdio>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
const int MAXN = 100000 + 10;
int n, k, fa[MAXN*3], ans, field;
int find(int x) {
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}
bool jud(int x, int y) {
    return find(x) == find(y);
}
void merge(int x, int y) {
    fa[find(x)] = find(y);
}
bool det(int x, int y, int ord) {
    if(x > n || y > n || (x == y && ord != 1)) 
        return false;
    if(ord == 1)  // x和y是同类
        if(jud(y + field, x) || jud(x + field, y)) 
            return false;
    if(ord == 2) // x吃y
        if(jud(y + field, x) || jud(x, y))  
            return false;
    return true;
}
int main() {
    scanf("%d %d", &n, &k);
    field = n + 1; // 自身 捕食 天敌
    for(int i=1; i<=MAXN*3-2; i++)
        fa[i] = i;
    for(int i=1; i<=k; i++) {
        int ord, x, y;
        scanf("%d %d %d", &ord, &x, &y);
        if(!det(x, y, ord)) {
            ans++;
            continue;
        }
        if(ord == 1) {
            merge(x, y);
            merge(x + field, y + field);
            merge(x + 2*field, y + 2*field);
        } else {
            merge(x + field, y);
            merge(x, y + 2*field);
            merge(y + field, x + 2*field);
        }
    }
    printf("%d\n", ans);
    return 0;
}