并查集例题：食物链

如果不懂并查集链接: 点这儿

题目

[NOI2001] 食物链

题目描述

动物王国中有三类动物 A,B,C，这三类动物的食物链构成了有趣的环形。A 吃 B，B 吃 C，C 吃 A。

现有 N 个动物，以 1 － N 编号。每个动物都是 A,B,C 中的一种，但是我们并不知道它到底是哪一种。

有人用两种说法对这 N 个动物所构成的食物链关系进行描述：

• 第一种说法是 1 X Y，表示 X 和 Y 是同类。
• 第二种说法是2 X Y，表示 X 吃 Y 。

此人对 N 个动物，用上述两种说法，一句接一句地说出 K 句话，这 K 句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

• 当前的话与前面的某些真的话冲突，就是假话
• 当前的话中 X 或 Y 比 N 大，就是假话
• 当前的话表示 X 吃 X，就是假话

你的任务是根据给定的 N 和 K 句话，输出假话的总数。

输入格式

第一行两个整数，N，K，表示有 N 个动物，K 句话。

第二行开始每行一句话（按照题目要求，见样例）

输出格式

一行，一个整数，表示假话的总数。

样例 #1

样例输入 #1

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

样例输出 #1

3

提示

1 ≤ N ≤ 5 ∗ 10^4

1 ≤ K ≤ 10^5

---

如何表示食物链中的关系呢

图中可以发现左图是普通的吃与被吃的关系，
右图中假设把一当为根节点，可以发现，与它距离为一的是被吃的，为二的是吃它的，为三的与它是同类。

但是 在问题中肯定不止123，所以对它与根节点的距离%3

查询根节点的函数

//f[i]是i的父节点,d[i]是i到根节点的距离
int find(int x)
{
	if (f[x] != x)
	{
		int t = find(f[x]);
		d[x] += d[f[x]];
		f[x] = t;
	}
	return f[x];
}

此为一个递归查询祖先的过程。由于根节点的父节点为它自己，结束条件就是 f[x]=x
注意： 不应先更新根节点，否则f[x]就变为了根节点

判断假话

最简单的就是样例中的第一个，总共100个动物，他来个101，肯定错
所以

if (a > n || b > n) ans ++;

so
当ab在同一集合中，ab关系要与事实相符，这里就可以用距离来判断了

if (fa == fb && (d[a] - d[b]) % 3) ans ++;
……
if (fa == fb && (d[a] - d[b] - 1) % 3) ans ++;

对于这题，如果当前这一种关系还没有在集合中，不能说明它是假的，只要把它加进去就好了

再此可以求出d[fa]

d[fa] = d[b] - d[a];

以上这是当说a与b是同类时
but
当a吃b时
两者距离%3应为1，所以

d[fa] = d[b] - d[a] + 1;

---

---

这就是整一道题的思路

下面给出完整代码

#include<bits/stdc++.h>
using namespace std;

int n, m, f[100010], a, b, d[100010], ans, c;

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

int main()
{
//freopen("a.in", "r", stdin);
//freopen("a.out", "w", stdout);
	cin >> n >> m;
	for (int i = 1; i <= n; ++ i) f[i] = i;
	while(m --)
	{
		cin >> c >> a >> b;
		if (a > n || b > n) ans ++;
		else
		{
			int fa = find(a), fb = find(b);
			if (c == 1)
			{
				if (fa == fb && (d[a] - d[b]) % 3) ans ++;
				else if (fa != fb)
				{
					f[fa] = fb;
					d[fa] = d[b] - d[a];
				}
			}
			else
			{
				if (fa == fb && (d[a] - d[b] - 1) % 3) ans ++;
				else if (fa != fb)
				{
					f[fa] = fb;
					d[fa] = d[b] - d[a] + 1;
				}
			}
		}
	}
	cout << ans;
	return 0;
}