acwing自我学习笔记-食物链 240-CSDN博客

本文链接：https://blog.csdn.net/qq_33596642/article/details/122810045

本文探讨了在使用并查集解决题目时，为何不能直接累加距离，而是需要先找到根节点。通过实例说明了`d[x]+=d[p[x]]`与`p[x]=Find(p[x])`的区别，以及如何确保正确计算到根节点的距离。重点在于优化路径压缩过程以提升算法效率。

摘要由CSDN通过智能技术生成

#include <iostream>

using namespace std;

const int N = 5e5+10;
int p[N],d[N];

int Find(int x)
{
   if (x != p[x])
   {
       int t = Find(p[x]);
       d[x] += d[p[x]];//到父结点的距离
       p[x] = t;
   }
   return p[x];
}

int main()
{
   int n, k;//n个动物,k句话
   cin >> n >> k;
   for (int i = 1; i <= n; i++)
   {
       p[i] = i;//初始化并查集
   }

   int D, x, y;
   int cnt = 0;//假话条数
   while (k--)
   {
       cin >> D >> x >> y;
       if ( x > n || y > n)
       {
           cnt++;
           continue;
       }
       int px = Find(x), py = Find(y);
       if (1 == D)//表明X,Y是同类
       {
           if (px == py && (d[x] - d[y])%3)//x,y在同一集合中，x,y到根结点的距离模3不同余，则不是同类
           {
               cnt++;
               continue;
           }
           else if (px != py)//不在同一个集合中，合并
           {
               p[px] = py;
               d[px] = d[y] - d[x];
           }
       }
       else//x吃y
       {
           if (px == py && (d[x] - d[y] - 1) % 3)
           {
               cnt++;
           }
           else if (px != py)
           {
               p[px] = py;
               d[px] = d[y] - d[x]+ 1;
           }
       }
   }
   cout << cnt << endl;
   return 0;
}