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 句话。

第二行开始每行一句话（按照题目要求，见样例）

输出格式
一行，一个整数，表示假话的总数。

输入输出样例
输入 #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

解题思路:
这题主要考察并查集相关知识
{需要掌握并查集基本方法如find(x)查找x的祖先节点,Union(x,y)将两个集合进行合并,这两个模板在代码中有体现,我也写了注释,就不再赘述!}
根据题意,1 X Y表示x和y是同类,那么(x,y)的猎物也应该是同类,(x,y)的天敌也应该是同类,2 X Y表示x吃y,那么x的猎物和y是同类,x的天敌应该是y的猎物!根据这个关系我们进行if语句判断,并且合并相同关系!
我们要开fa[3N]的数组(注意要开3倍哦),我们在1~n表示N个动物,n+1到2n表示N个动物对应的猎物,2n+1到3n表示N个动物的天敌！

代码:
(括号有点多,不过我都有备注,不用慌嘿嘿!)

#include<bits/stdc++.h>
#define maxn 150005
using namespace std;
int fa[maxn];
int ans;
int findFa(int x)
{
    //只有x的祖父节点的根节点才为本身
    if(x==fa[x])return x;
    //递归找跟节点,这里对并查集高度进行优化,变为2层不懂可以自己画个图!
    return fa[x]=findFa(fa[x]);
}

inline void Union(int x,int y)
{
    //先找到x,y的根节点,然后fa[x的根节点]指向y的根节点,即可合并成功!
    fa[findFa(x)]=findFa(y);
}
//判断是否在一个集合中
inline bool sameSet(int x,int y)
{
    return findFa(x)==findFa(y);
}

int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1; i<=3*n; i++)
    {
        fa[i]=i;
    }
    for(int i=1; i<=k; i++)
    {
        int c,x,y;
        scanf("%d%d%d",&c,&x,&y);
        if(x<=n&&y<=n)
        {
            //c==1,x和y是同类
            if(c==1)
            {
                //x和y的猎物,x和y的天敌都不是同一类,就Union他们,否则ans++
                if(!sameSet(x,y+n)&&!sameSet(x,y+2*n))
                {
                    Union(x,y);
                    Union(x+n,y+n);
                    Union(x+2*n,y+2*n);
                }
                else
                {
                    ans++;
                }
            }
            else  //c==2,x吃y
            {
                if(x!=y)
                {
                    //x和y不是同一类，x的天敌y+n和x也不是同一类
                    if(!sameSet(x,y)&&!sameSet(x,y+n))
                    {
                        //得到新关系
                        Union(x+n,y);
                        Union(x+2*n,y+n);
                        Union(x,y+2*n);
                    }
                    else
                    {
                        //x和y是同一类
                        ans++;
                    }
                }
                else
                {//x和y同类,与c不符合,ans++
                    ans++;
                }
            }
        }
        else//当输入的x或y大于N时越界,ans++
        {
            ans++;
        }
    }
    printf("%d",ans);

    return 0;
}

谢谢观看!