[Luogu] 并查集维护个数 P2078 朋友

最新推荐文章于 2022-06-19 17:01:31 发布

*DDL_GzmBlog

最新推荐文章于 2022-06-19 17:01:31 发布

阅读量194

点赞数

分类专栏： # 基础数据结构

本文链接：https://blog.csdn.net/qq_34364611/article/details/118071581

版权

基础数据结构专栏收录该内容

32 篇文章 0 订阅

订阅专栏

本文介绍了一种使用并查集数据结构解决实际问题的方法，针对两个同性别人群构成的公司社交网络，通过维护每个个体所属集合的大小，找出最可能的情侣组合数量。博主通过实例演示了如何运用并查集来简化情侣配对问题，核心在于理解问题的约束条件并巧妙利用数据结构。

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

(并查集维护集合元素个数模板题)

题意

两个公司 (全是同性别的人)

也就是一个公司如果有男生就不可能有女生

问你在小明(1代表) 和小红 (-1代表女生都用负数) 认识的人中

最可能凑出来几对情侣(包括他们自己)

吐槽
本来以为是组合数问题结果画了一下关系图就发现凑出来的情侣数由认识少的那一方限制
(那么问题就简单了 )

步骤

这不就是并查集维护集合中元素的个数问题嘛

那我们要怎么维护并查集集合中的个数呢
(其实很简单我们只要加一个 size数组即可在每次合并的操作的时候我们直接相加即可)

当然在初始化的时候每个size[i]都要初始化为1 毕竟本身也算嘛

if(fa!=fb)
{
    p[fa] = fb;
    sz[fb] +=sz[fa];
}

所以其他步骤就简单了就妥妥的并查集操作即可

CODE

#include <bits/stdc++.h>
#define IOS  ios::sync_with_stdio(false)
using namespace std;
const int N = 4e6+10;
int p[N],pm[N];
int n,m,x,q;
int sz[N],szm[N];

int find(int x)
{
    if(x != p[x])
    return p[x] = find(p[x]);
    return p[x];
}
int findm(int x)
{
    if(x != pm[x])
    return pm[x] = findm(pm[x]);
    return pm[x];
}

void init()
{
    for(int i=1; i<=N; i++)
    {
        p[i]  = i;
        pm[i] = i;
        sz[i] = 1;
        szm[i] = 1;
    }
}
void solve()
{
    cin>>n>>m>>x>>q;
    init();

    for(int i=1; i<=x; i++)
    {
        int a,b;
        cin>>a>>b;
        int fa = find(a);
        int fb = find(b);

        if(fa!=fb)
        {
            p[fa] = fb;
            sz[fb] +=sz[fa];
        }
    }
    for(int i=1; i<=q; i++)
    {
        int a,b;
        cin>>a>>b;
        a=-a, b=-b;
        int fa = findm(a);
        int fb = findm(b);

        if(fa!=fb)
        {
            pm[fa] = fb;
            szm[fb]+=szm[fa];

        }
    }


    int fa = find(1);
    int fb = findm(1);

    int ans = min(sz[fa],szm[fb]);
    cout<<ans<<endl;
}
int main()
{
    IOS;
    solve();
    return 0;
}