poj 1182 食物链(带权并查集)

题目：poj 1182 食物链(带权并查集)

中文题目，略。

解题思路：这题重点在于发现规律：如果x， y已经在同一个集合里的话，这样就有可能出现假话，(c【x】  + d - 1 + 3) != c[y] 这句话就是假话。这里的x吃y，那么c【y】  = 1，说明他与根x不同类且被根吃，相同则为0，为2，就是与根不同类但是吃根。

如果是不在同一个集合的话就需要合并成一个集合里，说明他们之间有关系并记录关系。合并： D X Y P 为X 的根，Q 为Y 的根，f【q】 = p；c【p】 = (c[x]  + d - 1 - c[y] + 3 ) % 3;

重点是这些规律要发现，这样这题就好做点。

代码：

#include <stdio.h>

const int N = 50005;
int n, m, f[N], c[N];

void init () {

	for (int i = 0; i <= n; i++) {

		f[i] = i;
		c[i] = 0;
	}
}

int getfather (int x) {

	if (x != f[x]) {

		int t = f[x];
		f[x] = getfather (f[x]);
		c[x] = (c[x] + c[t]) % 3;
	}
	return f[x];
}

int main () {
	
	scanf ("%d%d", &n, &m);
	init ();
	int x, y, z, count = 0;
	for (int i = 0; i < m; i++) {
		
		scanf ("%d%d%d", &z, &x, &y);
		if (x > n || y > n) {

			count++;
			continue;
		}
		if (z == 2 && x == y) {
			
			count++;
			continue;
		}
		int p = getfather (x);
		int q = getfather (y);
		if (p == q) {

			if ( (c[x] + z - 1 + 3) % 3 != c[y])
				count ++;
		} else {

			f[q] = p;
			c[q] = (c[x] - c[y] + z - 1 + 3 ) % 3; 
		}
	}
				
	printf ("%d\n", count);	
	return 0;
}