Vijos1776关押罪犯

题目链接：https://vijos.org/p/1776

 这个题是老师讲并查集的时候拿来的例题，记得以前做的时候是二分+二分图染色，这次竟然是贪心+并查集，所以就实现了一下。

整体思路可以参考这里，假设关在一起的是朋友（虽然有怨气值，但毕竟是比较少的），那么对于思路的实现间代码：

#include<iostream>
#include<vector>
#include<algorithm>
#define MAXN 20010
#define MAXM 100010
using namespace std;

struct Edge
{
    int x, y;
    long long w;
    Edge(int x, int y, int w):x(x), y(y), w(w) {}
};
int f[MAXN];  //并查集数组
int d[MAXN];  //i的敌对罪犯
vector<Edge> edges;

int cmp(Edge a, Edge b)
{
    return a.w > b.w;
}

int Find(int x) //并查集压缩路径
{
    if (f[x] == x)
        return x;
    f[x] = Find(f[x]);
    return f[x];
}

int main()
{
    int n, m;

    cin>>n>>m;
    for (int i=1; i<=n; i++) f[i] = i;
    for (int i=0; i<m; i++)
    {
        int x, y, w;
        cin>>x>>y>>w;
        edges.push_back(Edge(x,y,w));
    }

    sort(edges.begin(), edges.end(), cmp);
    int ans = 0;
    for (int i=0; i<edges.size(); i++)
    {
        Edge e = edges[i];
        if (d[e.x] == 0) //如果e.x还没有敌对罪犯，就为设当前敌对罪犯
            d[e.x] = e.y;
        if (d[e.y] == 0) ///
            d[e.y] = e.x;
        int x = Find(e.x);
        int y = Find(e.y);
        int dx = Find(d[e.x]);
        int dy = Find(d[e.y]);
        if (x == y)  //如果是朋友，则矛盾，退出
        {
            ans = e.w;   //本来这里是直接输出答案然后return 0的，但是莫名其妙只有一个点儿wrong answer，径队友建议改成现在的样子就全过了，也不知道为什么
            break;
        }
        if (x != y) //不是朋友，则合并相应的朋友
        {
            f[Find(x)] = Find(f[dy]);
            f[Find(dx)] = Find(f[y]);
        }
    }
    cout<<ans<<endl;

    return 0;
}