并查集的经典例题（三）

题目：营救

题目描述

妈妈下班回家，街坊邻居说小明被一群陌生人强行押上了警车！妈妈丰富的经验告诉她小明被带到了 t 区，而自己在 s 区。该市有 m 条大道连接 n 个区，一条大道将两个区相连接，每个大道有一个拥挤度。小明的妈妈虽然很着急，但是不愿意拥挤的人潮冲乱了她优雅的步伐。所以请你帮她规划一条从 s 至 t 的路线，使得经过道路的拥挤度最大值最小。

输入格式

第一行有四个用空格隔开的 n，m，s，t，其含义见【题目描述】。接下来 m 行，每行三个整数 u,v,w，表示有一条大道连接区 u 和区 v，且拥挤度为 w。两个区之间可能存在多条大道。

输出格式

输出一行一个整数，代表最大的拥挤度。

输入输出样例

输入：
3 3 1 3
1 2 2
2 3 1
1 3 3
输出：
2

解题思路

这道题跟之前的城市题有很大的相同点，都是道路连接城市，那我们不妨同样构造一个并查集。再次观察题目，题目需要我们寻找最大值中的最小，这时候就需要我们用到二分法。

代码

#include<iostream>
#define MAXN 10000
using namespace std;
int n,m,s,t,ans;
int u[MAXN],v[MAXN],w[MAXN],pre[MAXN];
//构造并查集
void init(){
	for(int i = 1;i <= n;++i)
		pre[i] = i;
}
//在并查集中查找元素 
int Find(int a){
	return a == pre[a] ? a : pre[a] = Find(pre[a]);
}
//寻找相等的最小值 
bool check(int mid){
	init();
	for(int i = 1;i <= m;++i){
		if(mid < w[i]) continue;
		int xx = Find(u[i]);
		int yy = Find(v[i]);
		if(xx != yy) pre[xx] = yy;
	}
	if(Find(s) == Find(t)) return true;
	else return false;	
} 

int main(){
	cin>>n>>m>>s>>t;
	int l = INT_MAX,r = INT_MIN;
	for(int i = 1;i <= m;++i){
		cin>>u[i]>>v[i]>>w[i];
		l = min(l,w[i]);
		r = max(r,w[i]);
	}
	//二分法进行筛选,不断地寻找最小值 
	while(l <= r){
		int mid = (l + r) / 2;
		if(check(mid)){
			r = mid - 1;
			ans = mid;
		}else
			l = mid + 1;
	}
	cout<<ans<<endl;
	return 0;
}