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
, c
;

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;
}