洛谷P1525： 关押罪犯（种类并查集 + 贪心）

https://www.luogu.org/problemnew/show/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≤1,000,000,000，且每对罪犯组合只出现一次。

 

输出格式：

 

共1 行，为Z 市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件，请输出0。

 

输入输出样例

输入样例#1： 复制

4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

输出样例#1： 复制

3512

说明

【输入输出样例说明】罪犯之间的怨气值如下面左图所示，右图所示为罪犯的分配方法，市长看到的冲突事件影响力是3512（由2 号和3 号罪犯引发）。其他任何分法都不会比这个分法更优。

【数据范围】

对于30%的数据有N≤15。

对于70%的数据有N≤2000,M≤50000。

对于100%的数据有N≤20000,M≤100000。

解题思路：

每个罪犯会只会在A和B两个监狱，那就用i表示第i个人在A监狱，i+n表示第i个人在B监狱，先按怨气值从大到小排序，

当x和y有仇恨时应该把他们放在两个监狱，有两种方法，一种是x+n和y即x去B，y去A，一种是x和y+n。

在合并之前要注意判断x和y是不是在同一监狱。

#include <stdio.h>
#include <algorithm>
#define N 100020
using namespace std;
struct date{
	int u, v, w;
}e[N];
int f[N];
int cmp(date a, date b)
{
	return a.w>b.w;
}
int getf(int v)
{
	if(f[v]==v)
		return v;
	f[v]=getf(f[v]);
	return  f[v];
}
void merge(int u, int v)
{
	int t1, t2;
	t1=getf(u);
	t2=getf(v);
	if(t1!=t2)
		f[t1]=t2;
} 
int main()
{
	int n, m, i, ans;
	scanf("%d%d", &n, &m);
	for(i=1; i<=2*n; i++)
		f[i]=i;
	for(i=0; i<m; i++)
		scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
	sort(e, e+m, cmp);
	ans = 0;
	for(i=0; i<m; i++)
	{
		if(getf(e[i].v)==getf(e[i].u) || getf(e[i].v+n)==getf(e[i].u+n))
		{
			ans = e[i].w;
			break;
		}
		merge(e[i].v, e[i].u+n);
		merge(e[i].v+n, e[i].u);
	}
	printf("%d\n", ans);
	return 0;
}