AcWing 257. 关押罪犯 (贪心 + 拓展域并查集)-CSDN博客

本文链接：https://blog.csdn.net/weixin_45682949/article/details/114788103

题目描述

分析

首先根据怨气值 $c$ 从大到小排序, 我们要求最小的怨气值, 所以对每一对有仇恨关系的罪犯,要尽可能的分到两座监狱中.
但是存在一种情况, 这一对罪犯 $A, B$ 必须位于同一座监狱:
即两座监狱中存在罪犯 $C$ 与这两名罪犯 $A, B$ 中的一位有仇恨关系.
而根据之前排序可知, $C$ 与 $A$ 或 $B$ 的怨气值一定大于 $A, B$ 的怨气值,则 $A, B$ 的怨气值 $c$ 就是答案(贪心)
而两座监狱, 我们可以维护拓展域并查集: 对于罪犯i,p[i]表示i所在监狱, p[i+N]表示i的敌人所在监狱

实现

#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 2e4 + 9, M = 1e5 + 9;
int n, m, p[2*N]; 
struct node
{
    int a, b, c;
};
node e[M];
bool cmp(node x, node y)
{
    return x.c > y.c;
}
void init()
{
    for(int i=0; i<2*n; i++) p[i] = i; // 别忘了将拓展域初始化
}
int find(int a)
{
    return p[a] == a ? a : p[a] = find(p[a]);
}
void merge(int a, int b)
{
    int fx = find(a);
    int fy = find(b);
    p[fx] = p[fy];
}
int main()
{
    cin >> n >> m;
    init();
    for(int i=0; i<m; i++)
    {
        cin >> e[i].a >> e[i].b >> e[i].c;
    }
    sort(e, e+m, cmp);
    int ans = 0;
    for(int i=0; i<m; i++)
    {
        int a = e[i].a;
        int b = e[i].b;
        int fx = find(a), fy = find(b);
        if(fx == fy) // a,b关在同一个监狱内，冲突不可避免，直接输出答案
        {
            ans = e[i].c;
            break;
        }
       
        merge(p[a],p[b + n]); // 将 a 与 b的敌人关在一起
        merge(p[b],p[a + n]); // 将 b 与 a的敌人关在一起
    }
    cout << ans << endl;
    return 0;
}