并查集例题（食物链）C++（Acwing）

最新推荐文章于 2024-03-24 15:38:15 发布

shady1972

最新推荐文章于 2024-03-24 15:38:15 发布

阅读量1k

点赞数 31

CC 4.0 BY-SA版权

分类专栏：算法文章标签： c++ 算法数据结构

本文链接：https://blog.csdn.net/qq_68772546/article/details/136257669

算法专栏收录该内容

38 篇文章

订阅专栏

这篇文章描述了一段C++代码，用于解决一个与并查集和路径压缩相关的算法问题。代码实现了一个find函数，用于处理树状数组并在给定条件下计算满足特定条件的元素对数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码：

#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", res);
    return 0;
}