1390：食物链【NOI2001】: (2种并查集)-queue.top

【题目描述】

动物王国中有三类动物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句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。

1）当前的话与前面的某些真的话冲突，就是假话；

2）当前的话中X或Y比N大，就是假话；

3）当前的话表示X吃X，就是假话。

你的任务是根据给定的N（1≤ N ≤50,000）和K句话（0≤K≤100,000），输出假话的总数。

【输入】

第一行是两个整数N和K，以一个空格分隔。

以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。

若D=1，则表示X和Y是同类。

若D=2，则表示X吃Y。

【输出】

只有一个整数，表示假话的数目。

【输入样例】

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

【输出样例】

3

这题就是一道并查集，用种类并查集或带权并查集都可以

（AC代码：）

做法一： 带权并查集

#include <bits/stdc++.h>
using namespace std;
const int N = 50010; 

int p[N], d[N];
int find(int x)
{
	if(x != p[x])
	{
		int u = find(p[x]);
		d[x] += d[p[x]];
		p[x] = u;
	}
	return p[x];
}
int main()
{
	int n, k;
	cin >> n >> k;
	int res = 0;
	for(int i = 1; i <= n; i ++) p[i] = i;
	while(k --)
	{
		int t, x, y;
		cin >> t >> x >> y;
		if(x > n || y > n)
		{
			res ++;
			continue;
		}
		int px = find(x), py = find(y);
		if(t == 1)
		{
			if(px == py && (d[x] - d[y]) % 3) res ++;
			else if(px != py)p[px] = py, d[px] = d[y] - d[x];
		}
		else 
		{
			if(px == py && (d[x] - d[y] - 1) % 3) res ++;
			else if(px != py) p[px] = py, d[px] = d[y] + 1 - d[x];
		}
	}
	cout << res;
	return 0;
}

做法二：种类并查集

#include<bits/stdc++.h>
using namespace std;
const int N = 150010;
int p[N];
int find(int x)
{
	if(p[x] != x) p[x] = find(p[x]);
	return p[x];
}
int main()
{
	int n, m;
	cin >> n >> m;
	for(int i = 1; i <= 3 * n; i ++) p[i] = i;
	int t, x, y, res = 0;
	for(int i = 1; i <= m; i ++)
	{
		cin >> t >> x >> y;
		if(x > n || y > n)
		{
			res ++;
			continue;
		} 
		if(t == 1) 
		{
			if(find(x) == find(y + n) || find(x + n) == find(y)) res ++;
			else
			{
				p[find(x)] = find(y);
				p[find(x + n)] = find(y + n);
				p[find(x + 2 * n)] = find(y + 2 * n);
			}
		}
		else
		{
			if(find(x) == find(y) || find(x + n) == find(y)) res ++;
			else
			{
				p[find(x)] = find(y + n);
				p[find(x + n)] = find(y + 2 * n);
				p[find(x + 2 * n)] = find(y);
			}
			
		} 
	}
	cout << res << endl; 
 	return 0;
}