AcWing 240. 食物链

#include <iostream>
using namespace std;
const int N = 50010;
int n, m;
int p[N], d[N];  //p是baba，d是距离
int find(int x) {
    if (p[x] != x) {//如果x不是树根
        int t = find(p[x]);//指向根节点   存起来  因为赋值完之后就变了
        d[x] += d[p[x]];//更新为到根节点之间的距离 ==到父节点之间的距离+父节点到根节点之间的距离
        p[x] = t;//再把赋值成根节点
    }
    return p[x];    
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) p[i] = i;//初始化，自己是一个集合
    int res = 0;//假话的个数
    while (m -- ) {
        int t, x, y;//t表示询问的种类
        scanf("%d%d%d", &t, &x, &y);
        if (x > n || y > n) res ++ ;//超出范围，假话
        else {
            int px = find(x), py = find(y);//找根节点
            if (t == 1) {  //如果是同类
                if (px == py && (d[x] - d[y]) % 3)//如果在一个集合而且距离%3不等
                    res ++ ;//假话
                else if (px != py) {//如果根节点不同，说明不在一个集合
                    p[px] = py;//把py变成px的根节点
                    d[px] = d[y] - d[x];//因为同一类，所以%3之后相等，所以距离直接为差
                }
            } else {//表示吃   如果x吃y，那么d[x]-d[y]=1
                if (px == py && (d[x] - d[y] - 1) % 3)//如果在一个集合， 而且不为1
                    res ++ ;//假话
                else if (px != py) {//如果不在一个集合
                    p[px] = py;//合并
                    d[px] = d[y] + 1 - d[x];
                }
            }
        }
    }
    printf("%d
", res);
    return 0;
}