洛谷P2024 食物链 【并查集】

题目描述

动物王国中有三类动物 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 句话，输出假话的总数。

标题输入格式

第一行两个整数，N，K，表示有 N 个动物，K 句话。
第二行开始每行一句话（按照题目要求，见样例）
输出格式
一行，一个整数，表示假话的总数。

这题是我在学习查并集的时候遇到的，所以想当然的用并查集来做了。

并查集内的每一个组表示组内所有元素发生或不发生

#include <iostream>
#include <algorithm>
#include <string>
#include <sstream>
#include <map>
#include <vector>
#include <set>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
int far[200000], srank[200000];
int finds(int x)//查找
{
    if (far[x] == x) {
        return x;
    }
    return far[x] = finds(far[x]);
}
//合并时rank小的向rank大的连边
void merges(int x, int y)//合并
{
    x = finds(x);
    y = finds(y);
    if (x == y) {
        return;
    }
    if (srank[x] < srank[y]) {
        far[x] = y;
    } else {
        far[y] = x;
        if (srank[x] == srank[y]) {
            srank[x]++;
        }//如果两个rank相等，则rank+1
    }
}
bool same(int x, int y)
{
    return finds(x) == finds(y);
}//判断是否在同一组
int main()
{
    int n, k, ans{0};
    cin >> n >> k;
    for (int i = 1; i<=3*n; i++) {
        far[i] = i;
        srank[i] = 0;
    }//初始化
    while (k--) {
        int a, x, y;
        cin >> a >> x >> y;
        if (x > n || y > n) {
            ans++;
            continue;
        }//x和y比n大，是假话
        if (a == 1) {
            if (same(x, y+n) || same(x, y+2*n)) {
                ans++;
                continue;
            }//x和y不是同类
            merges(x, y);
            merges(x+n, y+n);
            merges(x+2*n, y+2*n);
        } else {
            if (same(x, y) || same(x, y+2*n)) {
                ans++;
                continue;
            }//x和y是同类或者y吃x时，不成立
            merges(x, y+n);
            merges(x+n, y+2*n);
            merges(x+2*n, y);
        }
    }
    cout << ans <<endl;
    return 0;
}

来源：CSDN

作者：Embers_L

链接：https://blog.csdn.net/Embers_L/article/details/104010543