AcWing 257. 关押罪犯(并查集/扩展域或边带权)

题目链接

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。

数据范围

N≤20000,M≤100000

输入样例：

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

输出样例：

3512

思路：

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int N = 1e6 + 10;
const int M = 211;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
using namespace std;

int n,m;
int vis[N];
int ep[N];
int wp[N];
int np[N];
int tot;
int col[N];

void add(int a,int b,int c){
    ep[tot] = b;
    wp[tot] = c;
    np[tot] = vis[a];
    vis[a] = tot ++ ;
}

bool DFS(int x,int key,int k){
    col[x]=key;
    for(int i=vis[x];~i;i=np[i]){
        if(wp[i]<=k) continue;
        int j=ep[i];
        if(col[j]){
            if(col[j]==key) return 0;
        }
        else if(!DFS(j,3-key,k)) return 0;
    }
    return 1;
}

bool judge(int k){
    memset(col,0,sizeof(col));
    rep(i,1,n){
        if(!col[i]){
            if(!DFS(i,1,k)) return 0;
        }
    }
    return 1;
}

void solve(){
    cin>>n>>m;
    memset(vis,-1,sizeof(vis));
    while(m--){
        int x,y,z;
        cin>>x>>y>>z;
        add(x,y,z);
        add(y,x,z);
    }
    int l=0;
    int r=1e9;
    while(l<r){
        int mid=(l+r)>>1;
        if(judge(mid)) r=mid;
        else l=mid+1;
    }
    cout<<l<<endl;
}

int main() {
    solve();
	return 0;
}