vijos-P1531食物链

描述

动物王国中有三类动物 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。

输出格式

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

样例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

解题思路：此题为并查集的经典应用，将询问过的节点连接到根节点上，根据相对数值sign[]判断二者关系。

0代表同类，1代表吃根，2代表被根吃。

代码如下：

#include <iostream>
using namespace std;
#define MAXL 50050
int uf[MAXL], sign[MAXL];


void uf_init(int n);
int uf_find(int x);
void uf_union(int x, int y, int px, int py, int d);


int main()
{
	int n, k, d, x, y, lie = 0, px, py;
	cin >> n >> k;
	uf_init(n);
	for (int i = 0; i < k; i++)
	{
		cin >> d >> x >> y;
		if (x > n || y > n || (d == 2 && x == y))//排除异常情况
		{
			lie++;
			continue;
		}
		px = uf_find(x);
		py = uf_find(y);
		if (px == py)
		{
			if ((sign[y] + d + 2) % 3 != sign[x])//二者关系不正确
				lie++;
		}
		else
			uf_union(x, y, px, py, d);//两者不在同根，合并
	}
	cout << lie << endl;
	return 0;
}


void uf_init(int n)
{
	for (int i = 1; i <= n; i++)
	{
		uf[i] = i;
		sign[i] = 0;
	}
}


int uf_find(int x)
{
	if (uf[x] == x)	return x;
	int tmp = uf[x];
	uf[x] = uf_find(tmp);
	sign[x] = (sign[tmp] + sign[x]) % 3;
	return uf[x];
}
void uf_union(int x, int y, int px, int py, int d)
{
	uf[px] = py;
	sign[px] = (sign[y] - sign[x] + d + 2) % 3;
}

话说cin没有超时的说~~~