洛谷P2024 食物链(关于同余的证明)

题目描述

#食物链

题目描述

动物王国中有三类动物 $A,B,C$，这三类动物的食物链构成了有趣的环形。$A$ 吃 $B$，$B$ 吃 $C$，$C$ 吃 $A$。

现有 $N$ 个动物，以 $1\sim 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

样例输入 #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\le N\le 5\times 10^4$，$1\le K\le 10^5$。

对于本题个人认为有两个问题：
1.如何将并查集构建出来
2.如何处理取模的问题
我们依次来处理这两个问题：
对于问题一：因为题中有三类动物且会产生如下图所示关系

采用路径压缩的代码如下

int find(int x)
{
    if(p[x]!=x)
    {
        int t=p[x];             //存储父节点，因为路径压缩的时候父节点会变
        p[x]=find(p[x]);
        d[x]+=d[t];             //更新到新的父节点的距离，也就是祖宗节点
    }
}

然后来解决同类和捕食关系，同样用张图来表示

还有一个要要注意的问题就是图中x的正负，我们在问题二中仔细讲
问题2： 取模正负问题
我们在计算x的值时分别用

x=d[b]-d[a]
x=d[b]-d[a]+1

这两个公式难免会有x为负值的情况，在路径压缩之后就可能会出现一个点到祖宗节点的距离为负
导致出现错误，这里我举了个例子直观感受下

这里有两个解决方法，一是把两式相减看看是否为0，另一种就是人为的将取模运算改为正的
详情看代码

将取模改为正数

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;

const int N=50010;
int p[N],d[N];
int ans;
int n,k;

int find(int x)
{
    if(p[x]!=x)
    {
        int t=p[x];
        p[x]=find(p[x]);
        d[x]+=d[t]%3;
    }
    return p[x];
}

int main(void)
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) p[i]=i;
    
    int cnt=1;
    while(k--)
    {
        int t,a,b;
        scanf("%d%d%d",&t,&a,&b);
        if(a>n||b>n)
        {
            ans++;
            continue;
        }
        
        int ra=find(a),rb=find(b);
        if(t==1)
        {
            if(ra==rb)
            {
                if(d[a]%3==d[b]%3) continue;
                else ans++;
            }
            else
            {
                p[ra]=rb;
                d[ra]=((d[b]-d[a])%3+3)%3;
            }
        }
        else
        {
            if(ra==rb)
            {
                if((d[b]+1)%3==d[a]%3) continue;
                else ans++;
            }
            else
            {
                p[ra]=rb;
                d[ra]=((d[b]-d[a]+1)%3+3)%3;
            }
        }
    }
    
    printf("%d\n",ans);
    
    return 0;
}

将两式相减

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=50010;
int p[N],d[N];
int n,k,cnt;

int find(int x)
{
    if(p[x]!=x)
    {
        int t=find(p[x]);
        d[x]+=d[p[x]];
        p[x]=t;
    }
    return p[x];
}

int main(void)
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) p[i]=i;
    
    while(k--)
    {
        int a,x,y;
        scanf("%d%d%d",&a,&x,&y);
        if(x>n||y>n)
        {
            cnt++;
            continue;
        }
        
        int px=find(x),py=find(y);
        if(a==1)
        {
            if(px==py&&((d[y]-d[x])%3)) cnt++;
            else if(px!=py)
            {
                p[px]=py;
                d[px]=d[y]-d[x];
            }
        }
        else
        {
            if(px==py&&((d[y]+1-d[x])%3)) cnt++;
            else if(px!=py)
            {
                p[px]=py;
                d[px]=d[y]+1-d[x];
            }
        }
    }
    
    printf("%d",cnt);
    
    return 0;
}