洛谷P1525 —— 带权值并查集
题目链接:https://www.luogu.com.cn/problem/P1525
题目描述
S 城现有两座监狱,一共关押着 N 名罪犯,编号分别为 1 - N 。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久,如果客观条件具备则随时可能爆发冲突。我们用“怨气值”(一个正整数值)来表示某两名罪犯之间的仇恨程度,怨气值越大,则这两名罪犯之间的积怨越多。如果两名怨气值为 c 的罪犯被关押在同一监狱,他们俩之间会发生摩擦,并造成影响力为 c 的冲突事件。
每年年末,警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表,然后上报到 S 城 Z 市长那里。公务繁忙的 Z 市长只会去看列表中的第一个事件的影响力,如果影响很坏,他就会考虑撤换警察局长。
在详细考察了 N 名罪犯间的矛盾关系后,警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配,以求产生的冲突事件影响力都较小,从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨,那么他们一定会在每年的某个时候发生摩擦。
那么,应如何分配罪犯,才能使 Z 市长看到的那个冲突事件的影响力最小?这个最小值是多少?
输入格式
每行中两个数之间用一个空格隔开。第一行为两个正整数 N,M分别表示罪犯的数目以及存在仇恨的罪犯对数。接下来的 M 行每行为三个正整数 aj,bj,cj,表示 aj 号和 bj 号罪犯之间存在仇恨,其怨气值为 cj。数据保证1<aj≤bj≤N,0<cj≤1e9,且每对罪犯组合只出现一次。
输出格式
共 1 行,为 Z 市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件,请输出 0
。
代码:
1 #include <iostream> 2 #include <algorithm> 3 #include <cstring> 4 #include <cstdio> 5 6 using namespace std; 7 8 const int N = 20010, M = 1e5 + 10; 9 struct it 10 { 11 int a, b, w; 12 }its[M]; 13 int n, m; 14 int p[N]; 15 int d[N]; 16 17 bool cmp(it x, it y) 18 { 19 return x.w > y.w; 20 } 21 22 int find(int x) 23 { 24 if (p[x] != x) p[x] = find(p[x]); 25 return p[x]; 26 } 27 28 int main() 29 { 30 scanf("%d%d", &n, &m); 31 for (int i = 1; i <= n; i ++ ) p[i] = i; 32 for (int i = 1; i <= m; i ++ ) 33 { 34 int a, b, w; 35 scanf("%d%d%d", &a, &b, &w); 36 its[i] = {a, b, w}; 37 } 38 39 sort(its + 1, its + 1 + m, cmp); 40 41 int res; 42 for (int i = 1; i <= m + 1; i ++ ) 43 { 44 int fa = find(its[i].a), fb = find(its[i].b); 45 if (fa == fb) 46 { 47 res = its[i].w; 48 break; 49 } 50 else 51 { 52 if (!d[its[i].a]) d[its[i].a] = its[i].b; 53 else 54 { 55 int f = find(d[its[i].a]); 56 p[fb] = f; 57 } 58 if (!d[its[i].b]) d[its[i].b] = its[i].a; 59 else 60 { 61 int f = find(d[its[i].b]); 62 p[fa] = f; 63 } 64 } 65 } 66 67 cout << res << endl; 68 return 0; 69 }
本题,因为说了有“边权值”(我理解为矛盾值),所以要求出现矛盾情况下的最小边权值显然是需要排序的
那么问题又来了,我们要按照什么方法进行分配呢?
我们不妨这样想:两个人a,b有仇,那么把他们放在一起显然会打起来,那么我们还不如把a与b的其他敌人放在一起,
因为这样可能会出现“敌人的敌人就是朋友”的情况,恰好a与b的其他敌人之间没有矛盾,那么他们就可以放在同一个集合中,反之b对a亦然。
那么我们不妨这样实现: 首先需要并查集初始化
(1)先把所有的矛盾关系按照矛盾值从大到小排一遍序,
(2)接下来每次操作取出一个关系,看矛盾的两个人x和y是否已经分配到同一个集合中(并查集找父亲即可),那么还分如下两种情况:
如果在一起那么显然会打起来(会出现矛盾),那么直接输出当前的边权(矛盾值)即可(此时保证是最小矛盾值,因为已经排序了)
如果不在同一组,则按照“敌人的敌人就是朋友”的原则,把x与y的其他敌人分在同一组,y与x的其他敌人分在同一组
不断进行以上操作最终可以得到答案