给出一个n(n≤100)结点的图,求苗条度(最大边减最小边的值)尽量小的生成树。
【分析】
首先把边按权值从小到大排序。对于一个连续的边集区间[L, R],如果这些边使得n个点全部连通,则一定存在一个苗条度不超过W[R]-W[L]的生成树(其中W[i]表示排序后第i条边的权值)。
从小到大枚举L,对于每个L,从小到大枚举R,同时用并查集将新进入[L,R]的边两端的点合并成一个集合,与Kruskal算法一样。当所有点连通时停止枚举R,换下一个L(并且把R重置为L)继续枚举。
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 105, M = 5000, INF = 0x3f3f3f3f;
int n, m;
struct Edge{
int u, v, w;
}e[M];
bool cmp(Edge a, Edge b){
return a.w < b.w;
}
int f[N];
int find(int x){
return f[x] == x ? x : f[x] = find(f[x]);
}
void solve(){
int ans = INF;
sort(e, e+m, cmp);
for(int l = 0; l < m; l++){
for(int i = 1; i <= n; i++) f[i] = i;
int cnt = 0;
for(int r = l; r < m; r++){
int a = find(e[r].u);
int b = find(e[r].v);
if(a != b){
f[a] = b;
if( ++cnt == n-1){
ans = min(e[r].w-e[l].w, ans);
break;
}
}
}
}
if( ans == INF) ans = -1;
printf("%d\n", ans);
}
int main(int argc, char *argv[]) {
while(scanf("%d%d",&n,&m) == 2 && n){
for(int i = 0; i < m; i++)
scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
solve();
}
return 0;
}