UVA 1395 苗条的生成树（水题+Krustrl算法）

 

 

#include<iostream>
#include<algorithm>
#include<string>
#include<map>//int dx[4]={0,0,-1,1};int dy[4]={-1,1,0,0};
#include<set>//int gcd(int a,int b){return b?gcd(b,a%b):a;}
#include<vector>
#include<cmath>
#include<stack>
#include<string.h>
#include<stdlib.h>
#include<cstdio>
#define mod 1e9+7
#define ll long long
#define INF 100000000
#define ms memset
using namespace std;
const int maxn=1e4+5;

struct edge
{
    int x,y,w;
    edge(){}
};
edge e[maxn];
bool cmp(edge e1,edge e2)
{
    return e1.w<e2.w;
}
/*
题目大意:给定一张图，要求选建
一颗最小生成树使得最大边重和最小边重之差最小。

数据量太小了，直接暴力二重循环即可。
对每一次循环，遍历到刚好能建立生成树为止是最优的。（贪心）

*/
int n,m;
int pre[maxn];
int cnt=0,ans;

int Find(int x)
{
    return x==pre[x]?x:(pre[x]=Find(pre[x]));
}
bool Union(int x,int y)
{
    int tx=Find(x),ty=Find(y);
    if(tx==ty) return  false;
    pre[tx]=ty;
    return true;
}

int main()
{
    while(scanf("%d%d",&n,&m)&&(n||m))
    {
        for(int i=0;i<m;i++)  scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
        sort(e,e+m,cmp);

        ans=-1;
        for(int i=0;i<m;i++)
        {
            cnt=0;
            for(int t=1;t<=n;t++) pre[t]=t;

            for(int j=i;j<m;j++)
            {
                if(Union(e[j].x,e[j].y)==false) continue;
                if(++cnt==n-1)
                {
                    if(ans==-1) ans=e[j].w-e[i].w;
                    else   ans=min(ans,e[j].w-e[i].w);
                    break;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}