POJ 1182 食物链（带权并查集）

来源：互联网发布：刚哥哥淘宝店招制作编辑：程序博客网时间：2024/05/01 16:36

题目链接：食物链

这题是带权并查集的经典题，用rank[i]表示i与i父亲的关系，0为同类，1为吃，2为被吃。

rank在压缩路径和合并过程的转化在纸上把所有情况归纳出来，就可以变成一个公式。

代码：

#include <stdio.h>#include <string.h>const int N = 50005;int n, k, parent[N], rank[N], ans = 0;int find(int x) {if (parent[x] == x) return x;else {int p = find(parent[x]);rank[x] = (rank[x] + rank[parent[x]]) % 3;parent[x] = p;}return parent[x];}int main() {scanf("%d%d", &n, &k);for (int i = 1; i <= n; i++) {parent[i] = i; rank[i] = 0;}while (k--) {int q, a, b;scanf("%d%d%d", &q, &a, &b);if (a > n || b > n || (q == 2 && a == b)) ans++;else {int pa = find(a), pb = find(b);if (pa == pb) {if ((rank[b] - rank[a] + 3) % 3 != q - 1) ans++;}else {parent[pb] = pa;rank[pb] = (rank[a] - rank[b] + q + 2) % 3;}}}printf("%d\n", ans);return 0;}

1 0