简化问题的规定
x: 0, 表示同类。同类
y: 1, 表示y吃x。天敌
z: 2, 表示z吃y, x吃z。食物
find函数细节
int find(int x)
{
if (p[x] != x)
{
int t = find(p[x]);
d[x] += d[p[x]];
p[x] = t;
}
return p[x];
}
距离计算细节
int px = find(x), py = find(y);
if (t == 1)
{
if (px == py && (d[x] - d[y]) % 3) res ++ ;
else if (px != py)
{
p[px] = py;
d[px] = d[y] - d[x];
}
}
else if(t==2)
{
if (px == py && (d[x] - d[y] - 1) % 3) res ++ ;
else if (px != py)
{
p[px] = py;
d[px] = d[y] + 1 - d[x];
}
}
第一种说法是”1 X Y”,表示X和Y是同类。
(d[x] + ? -d[y]) % 3 == 0
d[px] = d[y] - d[x]
第二种说法是”2 X Y”,表示X吃Y。
(d[x] + ? +1 - d[y]) == 0
d[px] = d[y] + 1 - d[x]
#include<iostream>
using namespace std;
const int N = 5e4 + 10;
int p[N], d[N];
int find(int x) {
if (x != p[x]) {
int t = find(p[x]);
d[x] += d[p[x]];
p[x] = t;
}
return p[x];
}
int main() {
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) p[i] = i;
int res = 0;
while (m--) {
int t, x, y;
scanf("%d%d%d", &t, &x, &y);
if (x > n || y > n) res++;
else {
int px = find(x), py = find(y);
if (t == 1) {
if (px == py && (d[y] - d[x]) % 3) res++;
else if (px != py) {
p[px] = py;
d[px] = d[y] - d[x];
}
} else if (t == 2) {
if (px == py && (d[y] + 1 - d[x]) % 3) res++;
else if (px != py) {
p[px] = py;
d[px] = d[y] + 1 - d[x];
}
}
}
}
cout << res << endl;
return 0;
}