poj 3522(生成树)

题意：求解一个图中生成树最大边与最小边的最小差值

思路：循环去掉小边求残图的最小生成树，求得最小生成树的最长边与最短边的最小差值就是答案，容易证明与最小边差值最小的生成树最大边必然在最小生成树中。

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define inf 0x3f3f3f3f
#define N 110
#define M N*N
struct edge{
    int a,b,w;
}e[M];
int f[N];
int comp(const void *a,const void *b)
{
    return ((edge *)a)->w>((edge *)b)->w?1:-1;
}
int find(int x)
{
    if(f[x]==x||f[x]==0)return x;
    f[x]=find(f[x]);
    return f[x];
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m))
    {
        int min0=inf;
        if(n==0&&m==0)break;
        for(int i=0;i<m;i++)
        scanf("%d%d%d",&e[i].a,&e[i].b,&e[i].w);
        qsort(e,m,sizeof(edge),comp);
        for(int i=0;i<=m-n+1;i++)
        {
            int c=0;
            memset(f,0,sizeof(f));
            for(int j=i;j<m;j++)
            {
                int a0=find(e[j].a);
                int b0=find(e[j].b);
                if(a0==b0)continue;
                else {
                    f[a0]=b0;
                    c++;
                    if(c==n-1){min0=min(min0,e[j].w-e[i].w);break;}
                }
            }
            if(c<n-1)break;
        }
        if(min0<inf)printf("%d\n",min0);
        else printf("-1\n");
    }
    return 0;
}

关闭提示 关闭
确 认 取 消