洛谷2024食物链

题目描述

动物王国中有三类动物 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 中

一行，一个整数，表示假话的总数。

输入输出样例

输入样例#1：
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例#1：
3
说明

1 ≤ N ≤ 5 ∗ 10^4

1 ≤ K ≤ 10^5

---

知识点:并査集
对于已经确定关系的动物, 都加入并査集中.在维护并査集时,同时维护一个点到祖先的路径上点的权值之和, 用以记录捕食关系.具体操作见代码(自以为打得很清晰). 注意要 % 3. 想到思路以后就很容易做了.
以下是代码

#include<stdio.h>
using namespace std;
const int maxN = (int)5e4;
struct node
{
    int pre, val;
}a[maxN + 1];
void work(int x)
{
    if(a[x].pre == x)
        return;
    work(a[x].pre);
    a[x].val = (a[a[x].pre].val + a[x].val) % 3;
    a[x].pre = a[a[x].pre].pre;
}
int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i ++)
        a[i].pre = i, a[i].val = 0;
    int ans = 0;
    for(int i = 0; i < k; i ++)
    {
        int del, x, y;
        scanf("%d%d%d", &del, &x, &y);
        if(x > n || y > n || (del == 2 && x == y))
        {
            ans ++;
            continue;
        }
        del --;
        work(x);
        work(y);
        if(a[x].pre == a[y].pre)
            if((a[y].val - a[x].val + 3) % 3 != del)
            {
                ans ++;
                continue;
            }
        a[a[y].pre].val = (del - a[y].val + a[x].val + 3) % 3;
        a[a[y].pre].pre = a[x].pre;
    }
    printf("%d", ans);
}