UVa 1395 Slim Span

题意

有多组样例输入。
对于每组样例，第一行为两个整数$n(2\le n\le 100),m(0\le m\le n\ast (n-1)/2)$。
当$n$和$m$同时为$0$代表结束输入。
接下来有$m$行，每行三个整数$u,v,w$。
问当选取一些边来使得$n$个点联通时，最大边权-最小边权的差最小是多少，如果不能联通则输出$-1$。

思路

按边的权值从小到大排序，对每一条边先把其当做联通时权值最小的边，然后左端置为当前边的序号，右端设为边权值最大的边的序号。
再进行二分，利用并查集连边，如果边的下标取$l$到$mid$可以使$n$个点联通，就更新最小的答案。如果不存在联通的情况，$res$最终会是$inf$，最后判断一下如果$res$为$inf$则输出$-1$。

代码

#include <bits/stdc++.h>
#define inf 0x3f3f3f3f
#define ll long long
#define pi acos(-1.0)

using namespace std;

struct node{//边结点
	int u, v, val;
};

bool cmp(node a, node b){//按边的权值排序
	return a.val<b.val;
}

int n, m, u, v, val;
int f[105], cnt[105];
vector<node> vec;//存边

int find(int x){//寻根 
	return (f[x]==x ? x : f[x] = find(f[x]));
}

void join(int u, int v){//联通u和v
	int fu = find(u), fv = find(v);
	if (fu != fv){
		f[fu] = fv;
		cnt[fv] += cnt[fu];//记录联通块内点的个数
	}
}

bool solve(int s, int e){
	for (int i = 1; i <= n; i++){//对并查集进行初始化
		f[i] = i;
		cnt[i] = 1;
	}
	for (int i = s; i <= e; i++){
		int u = vec[i].u, v = vec[i].v;
		join(u, v);//联通u和v
	}
	return cnt[find(1)]==n ? true : false;//如果n个点联通为一块就返回true
}

int main(){
	while (scanf("%d%d", &n, &m) != EOF){
		if (n==0 && m==0){
			break;
		}
		int res = inf;
		while (m--){
			scanf("%d%d%d", &u, &v, &val);
			vec.push_back((node){u, v, val});//加边
		}
		sort(vec.begin(), vec.end(), cmp);//排序
		int sz = vec.size();
		for (int i = 0; i < sz; i++){
			int l = i, r = sz-1;//二分
			while (l <= r){
				int mid = (l+r)/2;
				if (solve(i, mid)){//如果可以联通
					res = min(res, vec[mid].val-vec[i].val);//更新res
					r = mid-1;
				}
				else{
					l = mid+1;
				}
			}
		}
		printf("%d\n", res==inf ? -1 : res);
		vec.clear();//因为多组样例，记得清空
	}
	
	return 0;
}