【并查集】食物链问题

食物链

Time Limit: 1000MS Memory Limit: 10000KTotal Submissions: 46219 Accepted: 13452

Description

动物王国中有三类动物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句话有的是真的，有的是假的。当一句话满足下列三条之一时，这句话就是假话，否则就是真话。 
1） 当前的话与前面的某些真的话冲突，就是假话； 
2） 当前的话中X或Y比N大，就是假话； 
3） 当前的话表示X吃X，就是假话。 
你的任务是根据给定的N（1 <= N <= 50,000）和K句话（0 <= K <= 100,000），输出假话的总数。 

Input

第一行是两个整数N和K，以一个空格分隔。 
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。 
若D=1，则表示X和Y是同类。 
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

100 71 101 1 2 1 22 2 3 2 3 3 1 1 3 2 3 1 1 5 5

Sample Output

3

#include <iostream>#include <cstdlib>#include <cstdio>using namespace std;class FoodLink {public:const static int N_MAX = 50000 + 10;const static int K_MAX = 100000 + 10;int t[N_MAX];// 表示关系t=1,表示子节点吃父节点；t=2表示父节点吃子节点int f[N_MAX];// 并查集数组fathervoid init() {//初始化所有f数组和t数组for (int i = 0; i < N_MAX; i++) {f[i] = i; t[i] = 0;}}int find(int x) {if (x == f[x]) return x;// 一级一级地累计好x节点的t值，并更新// 当find(x)完成后，x成为该并查集树的一级子节点int temp = f[x];f[x] = find(f[x]);// 需要注意的是，该并查集树的根节点，t值为0t[x] = (t[x] + t[temp]) % 3;return f[x];}void solve() {int D, N, K;int x, y;init();scanf_s("%d%d", &N, &K);int cnt = 0;for (int i = 0; i < K; i++) { scanf_s("%d%d%d", &D, &x, &y);// 去掉特殊情况if (x > N || y > N) { continue; }if (D == 2 && x == y) { continue; }int fx = find(x), fy = find(y);if (D == 1) {if (fx == fy && t[x] != t[y]) { continue; }cnt++; // 合并两个并查集树if (fx != fy) {f[fx] = fy;t[fx] = (t[fx] + 0 - (t[x] - t[y]) + 3) % 3;}}else { if (fx == fy && (t[x] - t[y] + 3) % 3 != 1) continue;cnt++;// 合并两个并查集树if (fx != fy) {t[fx] = (t[fx] + 1- (t[x] - t[y]) + 3) % 3;f[fx] = fy;}}}printf_s("%d\n", K-cnt);}};int main() {FILE *in, *out;FoodLink f;f.solve();return 0;}