当前位置: 代码迷 >> 综合 >> 洛谷OJ - P2024 - 食物链(并查集)
  详细解决方案

洛谷OJ - P2024 - 食物链(并查集)

热度:78   发布时间:2023-10-09 14:44:25.0
题目描述
动物王国中有三类动物 A,B,C,这三类动物的食物链构成了有趣的环形。A 吃 B,B吃 C,C 吃 A。现有 N 个动物,以 1 - N 编号。每个动物都是 A,B,C 中的一种,但是我们并不知道它到底是哪一种。有人用两种说法对这 N 个动物所构成的食物链关系进行描述:第一种说法是“1 X Y”,表示 X 和 Y 是同类。第二种说法是“2 X Y”,表示 X 吃 Y 。此人对 N 个动物,用上述两种说法,一句接一句地说出 K 句话,这 K 句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。? 当前的话与前面的某些真的话冲突,就是假话? 当前的话中 X 或 Y 比 N 大,就是假话? 当前的话表示 X 吃 X,就是假话你的任务是根据给定的 N 和 K 句话,输出假话的总数。
输入
从 eat.in 中输入数据第一行两个整数,N,K,表示有 N 个动物,K 句话。第二行开始每行一句话(按照题目要求,见样例)
输出
输出到 eat.out 中一行,一个整数,表示假话的总数。
样例输入
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
样例输出
3
说明

1 ≤ N ≤ 5 ? 10^41 ≤ K ≤ 10^5

题目思路

我们设[0,n)区间表示同类,[n,2*n)区间表示x吃的动物,[2*n,3*n)表示吃x的动物。

如果是关系1:
  将y和x合并。将y吃的与x吃的合并。将吃y的和吃x的合并。
如果是关系2:
  将y和x吃的合并。将吃y的与x合并。将y吃的与吃x的合并。
题目代码

#include <iostream>
#include <cstdio>
#include <algorithm> 
#define LL long long using namespace std;int p[150001]; int find(int x){return p[x] == x ? x : p[x] = find(p[x]);
}int n, k;
int a, x, y;
int ans = 0;
int main(){scanf("%d%d",&n, &k);for(int i = 0; i <= 3*n; i++) p[i] = i;for(int i = 0; i < k; i++){scanf("%d%d%d",&a, &x, &y);if(x<=0 || y<=0 || x>n || y>n) {ans++; continue;}if(a == 1){if(find(x+n) == find(y) || find(y+n) == find(x)) {ans++; continue;}p[find(x)] = find(y);p[find(x+n)] = find(y+n);p[find(x+2*n)] = find(y+2*n);}else{if(find(x) == find(y) || find(y+n) == find(x)) {ans++; continue;}p[find(x+n)] = find(y);p[find(y+2*n)] = find(x);p[find(x+2*n)] = find(y+n);}}printf("%d\n",ans);return 0;
}