题意:给你N,M,表示N个节点,M条边。接下来给出M条边。
求生成一棵树之后,树中最长的路和最短的路的差值最小。求出,若没有则输出-1.
思路:进行多次kruskal算法,每次找出差值,进行更新,最后输出差值。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <stack>
#include <map>
#include <iomanip>
#define PI acos(-1.0)
#define Max 5005
#define inf 1<<28
using namespace std;
int n,m;
struct kdq
{
int s,e,l;
}road[Max];
bool cmp(kdq a,kdq b)
{
return a.l<b.l;
}
int f[Max];
int find(int x)
{
return f[x]==x?x:f[x]=find(f[x]);
}
void Union(int x,int y)
{
y=find(y);
x=find(x);
if(x==y)return ;
if(y>x)
f[y]=x;
else
f[x]=y;
}
void init()
{
for(int i=0;i<=n+10;i++)
f[i]=i;
}
void kruskal()
{
int kk=inf;
for(int i=0;;i++)//进行多次kruskal
{
init();
int MAX=-inf,MIN=inf;
int num=0;
for(int j=i;j<m;j++)//成树过程
{
int a=find(road[j].s);
int b=find(road[j].e);
int w=road[j].l;
if(a!=b)
{
Union(a,b);
num++;
if(MAX<w)
MAX=w;
if(MIN>w)
MIN=w;
}
}
if(num==n-1)//如果还能成树
{
if(kk>MAX-MIN)//更新kk
kk=MAX-MIN;
if(kk==0)//如果kk为0了,那么直接跳出
break;
}
else
break;
}
if(kk==inf)//若kk没被更新,则输出-1
puts("-1");
else
printf("%d\n",kk);
}
int main()
{
int i,j,k,l;
while(scanf("%d%d",&n,&m),(n+m))
{
for(i=0;i<m;i++)
scanf("%d%d%d",&road[i].s,&road[i].e,&road[i].l);
sort(road,road+m,cmp);
kruskal();
}
return 0;
}