AcWing 257. 关押罪犯 (贪心 + 拓展域并查集)
作者:互联网
题目描述
分析
首先根据怨气值 c c c从大到小排序, 我们要求最小的怨气值, 所以对每一对有仇恨关系的罪犯,要尽可能的分到两座监狱中.
但是存在一种情况, 这一对罪犯 A , B A,B A,B必须位于同一座监狱:
即两座监狱中存在罪犯 C C C与这两名罪犯 A , B A,B A,B中的一位有仇恨关系.
而根据之前排序可知, C C C与 A A A或 B B B的怨气值一定大于 A , B A,B A,B的怨气值,则 A , B A,B A,B的怨气值 c c 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;
}
另一种做法
二分答案 + 染色法判断二分图 (个人觉得更好理解)
标签:监狱,node,int,fy,查集,257,罪犯,find,AcWing 来源: https://blog.csdn.net/weixin_45682949/article/details/114788103