P2078 朋友 - 情侣配对大作战题解-CSDN博客

本文链接：https://blog.csdn.net/qq_69849271/article/details/150355774

🌟 P2078 朋友 - 情侣配对大作战题解

题目链接

洛谷原题链接

题目描述

小明和小红想帮公司里的朋友们脱单！公司分为：

A公司：全是男生（编号为正整数）
B公司：全是女生（编号为负整数）

朋友关系具有传递性（朋友的朋友也是朋友）。小明（编号1）和小红（编号-1）是朋友。现在需要找出小明和小红朋友圈中能配成的最大情侣对数（情侣必须是一男一女）。

输入格式

第一行： $N$ （A公司男生数）， $M$ （B公司女生数）， $P$ （男生朋友关系数）， $Q$ （女生朋友关系数）
接下来 $P$ 行：每行两个正整数 $x, y$ ，表示男生 $x$ 和 $y$ 是朋友
接下来 $Q$ 行：每行两个负整数 $x, y$ ，表示女生 $x$ 和 $y$ 是朋友

输出格式

一个整数，表示最大情侣对数

算法思路

核心思想

使用两个并查集分别管理男生和女生的朋友圈：

男生圈：管理A公司男性朋友关系
女生圈：管理B公司女性朋友关系（负数转正数处理）

最大情侣对数 = min(小明朋友圈男生数, 小红朋友圈女生数)

关键步骤

初始化并查集：
- 男生圈： $N$ 个节点（编号1~ $N$ ）
- 女生圈： $M$ 个节点（编号1~ $M$ ，对应原负数编号绝对值）
合并朋友关系：
- 男生关系：直接合并
- 女生关系：取绝对值后合并
查询朋友圈大小：
- 小明朋友圈：男生圈中1号节点所在连通块大小
- 小红朋友圈：女生圈中1号节点所在连通块大小

复杂度分析

时间复杂度： $\alpha(N+M))$
- $\alpha$ 为反阿克曼函数（接近 $O (1)$ )
- $P$ ：男生关系数， $Q$ ：女生关系数
空间复杂度： $O (N + M)$

代码实现

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

const int MAXN = 10010; // 最大员工数

// 并查集结构体
struct UnionFind {
    int parent[MAXN];  // 父节点数组
    int size[MAXN];    // 连通块大小
    
    // 初始化并查集
    void init(int n) {
        for (int i = 1; i <= n; i++) {
            parent[i] = i;
            size[i] = 1;
        }
    }
    
    // 查找根节点（带路径压缩）
    int find(int x) {
        return parent[x] == x ? x : parent[x] = find(parent[x]);
    }
    
    // 合并两个节点（按秩合并）
    void unite(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);
        if (rootX == rootY) return;
        
        // 小树合并到大树
        if (size[rootX] < size[rootY]) 
            swap(rootX, rootY);
        
        parent[rootY] = rootX;
        size[rootX] += size[rootY];
    }
    
    // 获取连通块大小
    int getSize(int x) {
        return size[find(x)];
    }
};

int main() {
    int N, M, P, Q;
    cin >> N >> M >> P >> Q;
    
    // 初始化并查集
    UnionFind boyCircle, girlCircle;
    boyCircle.init(N); // 男生圈（A公司）
    girlCircle.init(M); // 女生圈（B公司）
    
    // 处理男生朋友关系
    while (P--) {
        int x, y;
        cin >> x >> y;
        boyCircle.unite(x, y);
    }
    
    // 处理女生朋友关系（负数转正数）
    while (Q--) {
        int x, y;
        cin >> x >> y;
        x = abs(x); 
        y = abs(y);
        girlCircle.unite(x, y);
    }
    
    // 获取朋友圈大小
    int boySize = boyCircle.getSize(1); // 小明朋友圈男生数
    int girlSize = girlCircle.getSize(1); // 小红朋友圈女生数
    
    // 最大情侣对数
    cout << min(boySize, girlSize) << endl;
    
    return 0;
}

代码解释

数据结构

UnionFind：并查集结构
- parent：存储每个节点的父节点
- size：存储每个连通块的大小
核心方法：
- init(n)：初始化 $n$ 个独立节点
- find(x)：查找根节点（路径压缩优化）
- unite(x,y)：合并两个节点（按秩合并优化）
- getSize(x)：获取 $x$ 所在连通块大小

关键逻辑

男生圈处理：

boyCircle.unite(x, y); // 直接合并男生关系

女生圈处理：

x = abs(x); y = abs(y); // 负数转正数
girlCircle.unite(x, y); // 合并女生关系

朋友圈大小计算：

int boySize = boyCircle.getSize(1); // 小明朋友圈大小
int girlSize = girlCircle.getSize(1); // 小红朋友圈大小

最大情侣对数：

min(boySize, girlSize); // 一男一女配一对

算法正确性证明

并查集优化

路径压缩：find操作时扁平化结构，使后续查询 $O (1)$
```
parent[x] = find(parent[x]); // 路径压缩
```
按秩合并：小树合并到大树，保证树高 $O(\log n)$
```
if (size[rootX] < size[rootY]) swap(rootX, rootY);
size[rootX] += size[rootY];
```