HDU 1829 A Bug's Life （并查集）

最新推荐文章于 2020-02-23 18:51:16 发布

原创最新推荐文章于 2020-02-23 18:51:16 发布 · 831 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #HDU #线段树 #并查集

本文介绍了一种使用并查集数据结构解决特定性别匹配问题的方法。通过维护两个数组par和pre，分别用于记录并查集的父节点和每个元素的前一个异性元素，实现对输入数据的有效处理。当遇到相同的性别时，利用并查集合并相关元素，并检查是否存在矛盾的情况。

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

这道题的思路不是自己想的，参考了别人的代码。数组par为并查集数组，pre为辅助数组。pre[i]为i个体之前对应的异性，与当前的异性属于同性，因此合并。然后就是查看是否同一个根节点啦~

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

int par[3000], pre[3000];
int n, m;

void init()
{
    for(int i = 1; i <= n; i++) par[i] = i;
}

int find1(int x)
{
    int r = x, j;
    while(r != par[r]) r = par[r];
    while(x != r) j = par[x], par[x] = r, x = j;

    return r;
}

void unite(int x, int y)
{
    x = find1(x);
    y = find1(y);
    par[x] = y;
}

int main()
{
    int t, a, b, x = 0;

    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);

        init();
        memset(pre, 0, sizeof pre);
        int f = 1;
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d", &a, &b);
            if(find1(a) == find1(b)) f = 0;

            if(pre[a] == 0) pre[a] = b;
            else unite(pre[a], b);

            if(pre[b] == 0) pre[b] = a;
            else unite(pre[b], a);
        }

        if(! f) printf("Scenario #%d:\nSuspicious bugs found!\n", ++x);
        else printf("Scenario #%d:\nNo suspicious bugs found!\n", ++x);
        printf("\n");
    }


    return 0;
}