扩展域并查集

扩展域并查集

---

扩展域并查集常用来维护多组关系的集合合并问题
比如维护两个有敌对关系的集合，就可以用扩展域并查集

食物链

---

首先我们定义一个物种a
a的天敌为a + n
a的猎物为a + n * 2
可以得出 a + n -> a -> a + n * 2
a + n * 2 -> a + n

#include <iostream>
using namespace std;
const int N = 5e5 + 10;
int p[N * 3], n, k, x, op, y, ans;
inline int find(int x)
{
	return x == p[x] ? x : p[x] = find(p[x]);
}
inline void add(int a, int b)
{
	if(find(a) != find(b))
	p[find(b)] = find(a);
}
int main()
{
	cin.tie(0) -> sync_with_stdio(0);
	cin >> n >> k;
	for(int i = 1; i <= n * 3; i++) p[i] = i;
	while(k--)
	{
		cin >> op >> x >> y;
		if(x > n || y > n)
		{
			ans++;
			continue;
		}
		if(op == 1)
		{
			if(find(x) == find(y + n) || find(x) == find(y + 2 * n))
			{
				ans++;
				continue;
			}
			add(x, y);
			add(x + n, y + n);//天敌并到一起
			add(x + 2 * n, y + 2 * n);//猎物并到一起
		}
		else {
			if(find(x) == find(y) || find(x) == find(y + n * 2) || x == y)//如果x,y属于一种物种或y的猎物是x
			{
				ans++;
				continue;
			}
			add(y + n, x); //y+n表示y的天敌
			add(x + n, y + n * 2);//x的天敌正好是y的猎物
			add(x + n * 2, y);//x的猎物是y
		}
	}
	cout << ans;
	return 0;
}