关押罪犯 - 并查集&优先队列

题目地址：http://www.51cpc.com/web/problem.php?id=4261

 

Summarize：

　此题最巧妙的是“敌人的敌人就是朋友！”，故需先将敌对关系放入优先队列，按怨恨值从大到小排序，建一个数组存储对应敌对关系，

　当遇到第二个以及后来的敌人时，将之前的敌人与后来的敌人放进一个监狱，直到遇见一对关系中两人已经关在统一监狱，即为最大值。

 

附代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;

#define LL long long
const int N = 2e4+5;
int n,m, f[N], enemy[N];
struct RELA {
    LL a, b, value;
    friend bool operator<(const struct RELA &x, const struct RELA &y) {
        return x.value<y.value;
    }
};

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

bool Union(int a, int b) {
    int x = find(a);
    int y = find(b);
    if(x == y) return false;
    if(x != y) f[x]=y;
    return true;
}

int main()
{
    ios::sync_with_stdio(false);
    
    while(cin>>n>>m)
    {
        LL a,b,value;
        priority_queue<RELA> rela;
        for(int i=1; i<=n; i++)
            f[i]=i, enemy[i]=0;
        for(int i=0; i<m; i++) {
            cin>>a>>b>>value;
            rela.push(RELA{a,b,value});
        }
        
        bool can=true;
        while(!rela.empty()) {
            RELA t = rela.top();
            rela.pop();
            a=t.a, b=t.b, value=t.value;
            
            if(!enemy[a]) enemy[a]=b;
            else Union(enemy[a],b);
            if(!enemy[b]) enemy[b]=a;
            else Union(enemy[b],a);
            
            if(find(a) == find(b)) {
                can=false;
                cout<<value<<endl;
                break;
            }
        }
        if(can) cout<<0<<endl;
    }
}