P1525 [NOIP2010 提高组] 关押罪犯

>>>将一些点分成两部分 使 矛盾的点数值最小

-》并查集 + 贪心

#include<cstdio>
#include<iostream>
#include<algorithm>
//#include<queue>
//#include<vector>
//#include<bits/stdc++.h>
#define ll long long
#define ddd printf("-----------------------\n");
using namespace std;
const int maxn=1e5+10 ;

struct edge{
    int x,y,z;
}e[maxn];
int n,m,cnt,fa[maxn],b[maxn];

bool cmp(edge a,edge b){
    return a.z>b.z;
}

int find(int x){
    return fa[x]==x? x:fa[x]=find(fa[x]);
}
void lian(int x,int y){
    int f1=find(x),f2=find(y);
    if(f1!=f2){
        fa[f1]=f2;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1;i<=n;i++) fa[i]=i;
    
    for(int i=1;i<=m;i++){
        int u,v,val;cin>>u>>v>>val;
        e[++cnt].x=u,e[cnt].y=v,e[cnt].z=val;
    }
    sort(e+1,e+1+cnt,cmp);
    for(int i=1;i<=cnt+1;i++)
    {
        if(find(e[i].x)==find(e[i].y)) { cout<<e[i].z<<'\n'; return 0;}
        else {
            if(b[e[i].x]==0) b[e[i].x]=e[i].y;
            else lian(b[e[i].x],e[i].y);
            if(b[e[i].y]==0) b[e[i].y]=e[i].x;
            else lian(b[e[i].y],e[i].x);
        }
        
    }
    return 0;
}