带边权并查集的实现原理和相关题目详解(AcWing 240. 食物链)

Summeeze

已于 2023-06-06 20:31:41 修改

阅读量140

点赞数 1

分类专栏：并查集文章标签：算法数据结构 c++ 动态规划

于 2023-04-13 23:01:00 首次发布

本文链接：https://blog.csdn.net/m0_73813117/article/details/130139420

版权

并查集专栏收录该内容

1 篇文章 0 订阅

订阅专栏

文章详细介绍了带边权并查集的实现原理，包括数组p和d的含义以及find函数的路径压缩功能。在AcWing240.食物链问题中，利用并查集处理动物之间的关系，通过节点到根节点的距离判断同类或捕食关系。文章还讨论了并查集合并时对d数组的特殊处理，以及如何根据题目需求调整合并策略。

摘要由CSDN通过智能技术生成

一. 带边权并查集的实现原理

并查集中数组的含义 :

p[i] 表示 i 的父节点
d[i] 表示 i 点距离父节点的距离

由上可知，p数组和d数组都在维护与父节点的关系，但我们在实际问题中通常需要通过维护它与根节点的关系来解决问题，因此我们需要通过find函数运用递归的思想实现路径压缩并且找到目标节点与根节点的关系。

find函数：

int find(int x)
{
    if (p[x] != x)
    {
        int root = find(p[x]);//1
        d[x] += d[p[x]];//2，x到父节点的距离+父节点到父节点的父节点的距离。
        p[x] = root;//3
    }
    return p[x];
}

短短几行代码，却包含了许多功能的实现。

通过第一行和第三行代码实现路径压缩，从x节点依次向上找到根节点，然后从上到下返回根节点的值，并且由p[x]接收，实现了从x到p[x]的路径上的所有节点的父节点均变为根节点，并且最终返回根节点。
通过第二行代码更新路径上的节点到新的父节点的距离，即根节点的距离。每次d[x]的值更新为x到父节点(注意此处的父节点指的是更新之前的父节点，如对于节点5，它的父节点为2，而不是1)的距离+父节点到父节点的父节点的距离，因为是递归实现，所以此时d[p[x]]中存的值为父节点到根节点的距离，因此d[x]更新为x点到根节点的距离。注意代码中第一行与第二行不能互换，因为我们运用了递归的思想更新d数组（从根节点依次向下），如果更改顺序，那么将变为由下至上更新，无法实现有效状态的转移。

并查集的合并：

   int pa = find(a), pb = find(b);
   if (pa != pb)
   {
       d[pa] = sz[pb];//1
       sz[pb] += sz[pa];//2，sz[x]为根节点为x的集合的节点个数。
       p[pa] = pb;//3
   }

在不同题目中，合并并查集时的操作有所不同，但关键点都在于对于根节点的d数组的赋值操作。以 AcWing 238. 银河英雄传说为例，若要将集合A与集合B合并，且A连接到B的后面，那么除了并查集的经典操作(第三行代码）之外，我们还要将A的根节点的d数组赋值为B集合的节点个数，代表pa到B的根节点的距离，在之后的过程中，只要我们调用find函数，如x为A集合中的一个节点，那么调用find（x）即可更新x到pa之间的所有节点的到B的根节点的距离。 —此过程类似于动态规划的状态转移，而对于根节点的赋值类似于动态规划的初始化。

二. AcWing 240. 食物链

链接

一. 题目

在这里插入图片描述

二. 分析

我们可以用每个节点到根节点的距离，表示每个节点和根节点的关系。规定当距离为1表示该节点吃根节点；为2时表示该节点被根节点吃；3表示与根节点为同类。那么对于X动物和Y动物来说：

1.定义X和Y为同类时

若X与Y在同一集合中，( d[X] - d[Y] ) % 3 = 0，为真话，不为0为假话。
不在一个集合内，则将X所在集合连接到Y所在集合上，设将X的根节点的d数组赋值为eor，为保证X和Y为同类，则d[X] + eor = d[Y]，推出：d[PX] = d[Y] - d[X] ；

2.定义X吃Y时

若X与Y在同一集合中，( d[X] - d[Y] - 1) % 3 = 0，为真话，不为0为假话。
不在一个集合内，则将X所在集合连接到Y所在集合上，设将X的根节点的d数组赋值为eor，为保证X吃Y，则d[X] + eor - 1 = d[Y]，推出：d[PX] = d[Y] - d[X] + 1；

三. 代码

#include <iostream>

using namespace std;

const int N = 50010;

int n, m;
int p[N], d[N];

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()
{
    scanf("%d%d", &n, &m);

    for (int i = 1; i <= n; i ++ ) p[i] = i;

    int res = 0;
    while (m -- )
    {
        int t, x, y;
        scanf("%d%d%d", &t, &x, &y);

        if (x > n || y > n) res ++ ;
        else
        {
            int px = find(x), py = find(y);
            if (t == 1)
            {
                if (px == py && (d[x] - d[y]) % 3) res ++ ;
                else if (px != py)
                {
                    p[px] = py;
                    d[px] = d[y] - d[x];
                }
            }
            else
            {
                if (px == py && (d[x] - d[y] - 1) % 3) res ++ ;
                else if (px != py)
                {
                    p[px] = py;
                    d[px] = d[y] + 1 - d[x];
                }
            }
        }
    }

    printf("%d\n", res);

    return 0;
}