食物链

题目

动物王国中有三类动物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),输出假话的总数。 

Input

第一行是两个整数N和K,以一个空格分隔。 
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。 
若D=1,则表示X和Y是同类。 
若D=2,则表示X吃Y。

Output

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

样例

Sample Input

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

Sample Output

3

用0  1   2 分别表示A B C的关系。0吃1,1吃2,2吃0.注意这个编号都是以根结点为参照的,不是绝对的。开一个val数组,一开始这个数组为0,所有的点都是独立的,不是相连的,没有关系。慢慢加入点之后,把有关系的合并在一起,并且编号的相对大小确定一个集合中的关系。这题比较坑的地方就是,一定要单组输入,否则会WA。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
typedef long long ll;
#define INF 0x3f3f3f3f
using namespace std;
#define maxn 50005
int par[maxn];
int val[maxn];
int find(int x) {
	if (par[x] == -1) return x;
	int tmp = find(par[x]);
	val[x] += val[par[x]];
	val[x] %= 3;
	return par[x] = tmp;
}
int main() {
	//ios::sync_with_stdio(false);
	int n, m;
	int u, v, w;
	scanf("%d%d", &n, &m);
	int ans = 0;
	memset(par, -1, sizeof(par));
	memset(val, 0, sizeof(val));
	while (m--) {
		scanf("%d%d%d", &u, &v, &w);
		if (v > n || w > n) {
			ans++;
			continue;
		}
		int x = find(v);
		int y = find(w);
		if (x == y) {
			if (u == 1 && val[v] != val[w]) ans++;
			if (u == 2 && (val[v] + 1) % 3 != val[w]) ans++;
		}
		else {
			if (u == 1) {
				par[y] = x;
				val[y] = val[v] - val[w];
				val[y] = (val[y] + 3) % 3;
			}
			else {
				par[y] = x;
				val[y] = val[v] - val[w] + 1;
				val[y] = (val[y] + 3) % 3;
			}
		}
	}
	cout << ans << endl;
	return 0;
}

 

你可能感兴趣的:(递归与并查集,并查集)