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

参考博客：POJ-1182 食物链
思路：经典的带权并查集。除了并查集那个数组外，多开一个权值数组，或者把两个合成一个结构体数组。权值为每个点与根节点的关系//0:同类 1:吃 2:被吃。
图一：路径压缩 
图二：合并操作
图三：a和b相对关系



//#include<bits/stdc++.h>
#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN = 50005;
int n, m, ans, f[MAXN], w[MAXN];//并查集数组和关系数组
//0:同类 1:吃 2:被吃
void init()
{
ans = 0;
for (int i = 0; i <= n; i++)
{
f[i] = i; w[i] = 0;
}
}
int Find(int x)
{
if (f[x] == x) return x;
int t = f[x];
f[x] = Find(f[x]);
w[x] = (w[x] + w[t]) % 3;
return f[x];
}
void Union(int a, int b, int D)
{
int root1 = Find(a), root2 = Find(b);
if (root1 != root2)
{
f[root1] = root2;
w[root1] = (w[b] + (D - 1) - w[a] + 3) % 3;
}
else
{
if ((w[a] - w[b] + 3) % 3 != D - 1) ans++;
}
}
int main()
{
//while(~scanf("%d%d", &n, &m))不知道为什么用多组输入输出会wa
scanf("%d%d", &n, &m);
{
init();//初始化
while (m--)
{
int D, x, y;
scanf("%d%d%d", &D, &x, &y);
if (x > n || y > n || (D == 2 && x == y)) { ans++; continue; }//大于N或者自己吃自己
Union(x, y, D);
}
printf("%d\n", ans);
}
return 0;
}
/*
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
*/