UVA-1395（最小瓶颈生成树）

题目地址戳我

题意：多组数据，每组数据第一行给出N,M代表点和边的数量。接着M行每行u,v,w代表u和v之间的权值为w（有向无向貌似不影响），对于图中的每个生成树，求出生成树中最大权值与最小权值的差，求出所有生成树中最大权值与最小权值差的最小值。

思路：对权值排序后枚举L~R的区间即可。

代码

#include<stdio.h>
#include<string.h>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
//最大生成树
const int maxn=5005;
const int INF=0x3f3f3f3f;
struct node
{
    int u;
    int v;
    int w;
} edge[maxn];
int Father[maxn];//并查集数组
int N;
int M;
void Init_Father()
{
    for(int i=0; i<=N; i++)
        Father[i]=i;
}
int Find_Father(int x)//返回x的父结点
{
    if(x!=Father[x])
        Father[x]=Find_Father(Father[x]);//路径压缩
    return Father[x];
}
bool Merge_Father(int x,int y)//判断x和y是否在同一集合
{
    int flag_x=Find_Father(x);
    int flag_y=Find_Father(y);
    if(flag_x!=flag_y)
    {
        Father[flag_x]=flag_y;
        return true;//不在同一集合，就合并
    }
    return false;//在同一集合，就返回不可合并
}
bool cmp(node x,node y)
{
    return x.w<y.w;//按权值升序排列
}
int main()
{
    while(~scanf("%d%d",&N,&M))
    {
        if(N==0&&M==0)
            break;
        for(int i=0; i<M; i++)
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        sort(edge,edge+M,cmp);
        int result=INF;//最终结果
        for(int L=0; L<M; L++)
        {
            int num_edge=0;
            Init_Father();
            for(int R=L; R<M; R++)
            {
                if(Merge_Father(edge[R].u,edge[R].v))
                {
                    num_edge++;
                }
                if(num_edge==N-1)
                {
                    result=min(result,edge[R].w-edge[L].w);
                }
            }
        }
        if(result==INF)
            printf("-1\n");
        else
            printf("%d\n",result);
    }
    return 0;
}