POJ 3522 Slim Span MST变形

题意：给你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;
}