P2024食物链题解（详细讲解距离数组d在并查集中的应用）

       在并查集的题目中，出题人很喜欢将并查集与元素到父亲节点的距离数组d放在一起考察，而能够理解在寻找父节点时d数组的更新过程，对于思考问题有很大的帮助，本篇文章以这道经典题目为例讲解d数组对于求解并查集的帮助。

题目链接：[NOI2001] 食物链 - 洛谷

# [NOI2001] 食物链

## 题目描述

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

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

       题意翻译：动物园中有三种动物形成一个吃与被吃的关系，题目给出了一些句子，其中包含同类与捕食的关系，我们需要判断有多少是假话

       思考：首先我们看到题目中动物分为好几类，我们首先想到了并查集，并查集的最直观想法是三种动物，就将三种动物分为三个集合，但是这样分的话，我们怎么判断动物之间捕食的关系呢，同时一个动物属于哪个集合我们一开始也并不确定，因此这种想法果断否决了。

我们发现动物之间有捕食的关系，因此它们三者之间有一个距离的关系如下所示：

 同时为了方便判断一个动物是否已经确定捕食关系，我们可以把动物分为两类：

1、已经确定了与其他动物的捕食关系

2、还未确定与其他动物的捕食关系

这样就可以用并查集维护已经确定关系的动物，同时通过他们之间的距离关系判断他们的捕食关系，也就是说，如果动物a与动物b满足（d[a] - d[b]）%3==0，那么他们就是同类，否则就存在捕食关系

确定大致做法后，我们要着重讲一下并查集中距离数组的更新，很重要：

首先我们一定要清楚：d数组表示当前该节点距离当前父节点的距离，不然后面会混乱

 上图表明，寻找父节点是递归调用的过程，这也就是为什么寻找父节点的数组是这样写的原因：

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

这里理解之后，下面就是要解决在发现一对新的关系后，父亲节点距离数组的更新问题，还是分情况讨论：

1、a与b属于同类，那么他们满足（d[a] - d[b]) %3 = 0

2、a捕食b，那么a到根节点距离比b多1，所以（d[a] - d[b] - 1）%3 = 0

所以父节点距离更新如下图：

 

 下面附上完整源代码：

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

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

int find(int x){
    if(x!=p[x]){
        int u=find(p[x]);   
        //加上更新后的距离
        d[x]+=d[p[x]];
        p[x]=u;
    }
    return p[x];
}

int main(){
    cin>>n>>k;
    
    for(int i=1;i<=n;i++) p[i]=i;
    
    int res=0;
    while(k--){
        int op,a,b;
        scanf("%d%d%d",&op,&a,&b);
        //编号不满足情况
        if(a>n||b>n) res++;
        else {
            int fa=find(a),fb=find(b);
            if(op==1){
                if(fa==fb&&(d[a]-d[b])%3) res++;
                else if(fa!=fb) {
                    p[fa]=fb;
                    d[fa]=d[b]-d[a];
                }
            }
            else if(op==2) {
                if(fa==fb&&(d[a]-1-d[b])%3) res++;
                else if(fa!=fb) {
                    p[fa]=fb;
                    d[fa]=d[b]+1-d[a];
                }
            }
        }
    }
    
    cout<<res<<endl;
    
    return 0;
}

下面附上一道也与距离数组有关的并查集问题，同时还要维护每个集合的大小

[NOI2002] 银河英雄传说 - 洛谷