题目链接:##
题目描述:##
扩展域并查集
三种动物会构成一个食物环,可参考石头剪刀布
由于对于一个(x)有三种关系:
- 同类
- 食物
- 天敌
所以将一个(x)拆分成三个域,然后对于两种操作分别合并同类项
对于(x)和(y)是同类的情况,直接一一对应合并两者的三个域即可。
对于(x)吃(y)的情况,将(x)的同类并(y)的天敌,(x)的食物并(y)的同类,(x)的天敌并(y)的食物
对于(x)和(y)是同类的操作,有两类关系是不合法的:
- (x)吃(y)
- (y)吃(x)
对于(x)吃(y)的操作,有两类关系是不合法的: - (x)和(y)是同类
- (y)吃(x)
代码:##
#include<bits/stdc++.h>
#define N 3 * (50000)
using namespace std;
inline int read() {
int cnt = 0, f = 1; char c;
c = getchar();
while (!isdigit(c)) {
if (c == '-') f = -1;
c = getchar();
}
while (isdigit(c)) {
cnt = c - '0' + cnt * 10;
c = getchar();
}
return cnt * f;
}
int fa[N], n, k, cnt = 0, opr;
int x, y;
int get_father(int x) {
if(fa[x] == x) return x;
return fa[x] = get_father(fa[x]);
}
bool check(int opr, int x, int y) {
if(x > n || y > n) return false;
int x1 = get_father(x), x2 = get_father(x + n), x3 = get_father(x + 2 * n);
int y1 = get_father(y), y2 = get_father(y + n), y3 = get_father(y + 2 * n);
if (opr == 1) {
if(x2 == y1) return false;
if(x1 == y2) return false;
}
if (opr == 2) {
if(x == y) return false;
if(x1 == y1) return false;
if(x1 == y2) return false;
}
return true;
}
bool merge(int opr, int x, int y) {
int x1 = get_father(x), x2 = get_father(x + n), x3 = get_father(x + 2 * n);
int y1 = get_father(y), y2 = get_father(y + n), y3 = get_father(y + 2 * n);
if (!check(opr, x, y)) return false;
if (opr == 1) {
fa[y1] = x1; fa[y2] = x2; fa[y3] = x3;
}
if (opr == 2) {
fa[y1] = x2; fa[y2] = x3; fa[y3] = x1;
}
return true;
}
int main() {
n = read(); k = read();
for (register int i = 1; i <= 3 * n; i++) fa[i] = i;
for (register int i = 1; i <= k; i++) {
opr = read(); x = read(); y = read();
cnt += !merge(opr, x, y);
}
printf("%d", cnt);
return 0;
}