食物链（并查集维护动物关系 + 路径压缩优化）

 240. 食物链

输入样例：

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

题目链接：https://www.acwing.com/problem/content/description/242/

 代码展示：

#include <iostream>

using namespace std;

const int N = 500010;
int p[N], d[N];

int find (int x)//找祖宗并记录到祖宗的距离+优化:路径压缩
{
    if (p[x] != x)
    {
        int t = find(p[x]);
        d[x] += d[p[x]];//d[x]到p[x]的距离+p[x]到祖宗的距离
        p[x] = t;
    }
    return p[x];
}
int main()
{
    int n, k, res = 0;//res记录假话个数
    scanf("%d%d",&n,&k);
    //初始化
    for (int i = 1; i <= n; i ++ ) p[i] = i;
    
    while (k --)
    {
        int op, x, y;
        scanf("%d%d%d",&op,&x,&y);
        if (x > n || y > n) res ++;
        else
        {
            int px = find(x), py = find(y);//px,py为x,y的祖宗
            if (op == 1)
            {
                if (px == py && (d[x] - d[y]) % 3 != 0) res ++;//px和py在一个树里，x,y到祖宗的距离不同余
                else if (px != py)//不在一棵树，就要连成一棵树
                {
                    p[px] = py;//把x的祖宗连在y所在祖宗
                    d[px] = d[y] - d[x];//同类
                }
            }
            else
            {
                if (px == py && (d[x] - d[y] - 1) % 3  != 0) res ++;
                else if (px != py)
                {
                    p[px] = py;
                    d[px] = d[y] - d[x] + 1;
                }
            }
        }
    }
    printf ("%d\n",res);
}