关押罪犯题解【并查集】

题目：

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<=1000000000,且每对罪犯组合只出现一次。

样例输出：

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

题解：

首先对c进行从大到小排序，把怨气值较大的两个人处理在不同的两个集合。
其次开始并集，如果输入的两个人：a，b在同一个集合就说明他们会产生冲突，输出c，然后跳出即可。如果在不同的集合那么就要将a，b分别并到对方的敌对集合，总之要让两人尽可能并到不同集合，使得两人不产生冲突。那么如何建立敌对集合呢？，你可以将数组加个n，n+i表示i的敌对集合即可。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int fa[200005], rank[200005];
void MakeSet(int n) {
    for (int i = 1; i <= n; i++) {
        fa[i] = i;
        rank[i] = 0;
    }
}
int FindSet(int x) {
    if (fa[x] != x) {
        fa[x] = FindSet(fa[x]);
    }
    return fa[x];
}
void UnionSet(int w, int e) {
    int u = w, v = FindSet(e);
    if (u == v) {
        return;
    }
    if (rank[u] > rank[v]) {
        fa[v] = u;
    } else {
        fa[u] = v;
        if (rank[u] == rank[v]) {
            rank[u]++;
        }
    }
}
struct node {
    int a, b, c;
} zui[10000005];
int cmp(node x, node y) { return x.c > y.c; }
int main() {
    int n, m, q, x, y;
    scanf("%d %d", &n, &m);
    MakeSet(n * 2);
    for (int i = 1; i <= m; i++) {
        scanf("%d %d %d", &zui[i].a, &zui[i].b, &zui[i].c);
    }
    int sum = 0;
    sort(zui + 1, zui + 1 + m, cmp);
    for (int i = 1; i <= m; i++) {
    	int o = FindSet(zui[i].a), p = FindSet(zui[i].b);
        if (o == p) {//如果在同一集合产生冲突
            printf("%d", zui[i].c);
            return 0;
        }
        UnionSet(o, zui[i].b + n);
        UnionSet(p, zui[i].a + n);//否则分别加到敌对集合
    }
    printf("0");
    return 0;
}