关押罪犯(贪心,并查集)

题目链接:https://www.acwing.com/problem/content/259/

题目
$S$ 城现有两座监狱，一共关押着 $N$ 名罪犯，编号分别为 $1\sim N$。

他们之间的关系自然也极不和谐。

很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。

我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。

如果两名怨气值为 $c$ 的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为 $c$ 的冲突事件。

每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到 $S$ 城 $Z$ 市长那里。

公务繁忙的 $Z$ 市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。

在详细考察了 $N$ 名罪犯间的矛盾关系后，警察局长觉得压力巨大。

他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。

假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。

那么，应如何分配罪犯，才能使 $Z$ 市长看到的那个冲突事件的影响力最小？这个最小值是多少？

输入格式
第一行为两个正整数 $N$ 和 $M$，分别表示罪犯的数目以及存在仇恨的罪犯对数。

接下来的 M 行每行为三个正整数 $a_j$，$b_j$，$c_j$，表示 $a_j$ 号和 $b_j$ 号罪犯之间存在仇恨，其怨气值为 $c_j$。

数据保证 1≤aj<bj<N,0<cj≤109 且每对罪犯组合只出现一次。

输出格式
输出共 $1$ 行，为 $Z$ 市长看到的那个冲突事件的影响力。

如果本年内监狱中未发生任何冲突事件，请输出 $0$。

数据范围
$N\le 20000,M\le 100000$

输入样例：

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

输出样例：

3515

思路:

先按怨气值从高到低排序
每次将 $a$ 与 $b+n$ 合并， $b$ 与 $a+n$ 合并 表示 $a$ 和 $b$ 关在两个监狱
如果 $a$ 和 $b$ 在一棵树上，说明 $a$ 和 $b$ 关在一个监狱，此时的怨气值就是答案

AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 20010,M = 100010;
int p[N * 2];

int n,m;
bool flag = true;

struct Node{
    int a;
    int b;
    int c;
}arr[M];

int find(int x)
{
    if(p[x] != x)
    {
        p[x] = find(p[x]);
    }
    return p[x];
}


bool cmp(const Node &a,const Node &b)
{
    return a.c > b.c;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 0; i < m; i++) scanf("%d%d%d",&arr[i].a,&arr[i].b,&arr[i].c);
    sort(arr,arr+m,cmp);
    for(int i = 1; i <= 2 * n; i++) p[i] = i;
    for(int i = 0; i < m; i++)
    {
        int a,b,pa,pb;
        a = arr[i].a;
        b = arr[i].b;
        p[find(a)] = find(p[b + n]); // a和b+n匹配
        p[find(b)] = find(p[a + n]); // b和a+n匹配
        pa = find(a);
        pb = find(b);
        if(pa == pb)
        {
            flag = false;
            printf("%d\n",arr[i].c);
            break;
        }
    }
    if(flag) printf("0\n");
    return 0;
}