poj3522 并查集+kruskal_poj2832 并查集-优快云博客

本文链接：https://blog.youkuaiyun.com/karen_mo/article/details/7786325

本文介绍了作者在解决poj3522问题时，最初尝试使用prim算法但未通过测试，后来转向使用kruskal算法实现。通过排序和枚举构建最小生成树，以达到题目要求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个题目我一开始的思路是利用prim算法，选出边数最少的点，利用循环，每次选出一条边进行枚举，选出最大值与最小值差距最小的生成树。

但是写完之后交上去怎么都过不了，于是也按照网上的主流算法，利用kruscal算法来写。

先进行排序，再进行枚举。。。

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define MAXN 105
#define MAXM 10050
#define INF 99999999
#define MIN(a,b) a>b?b:a
struct Node
{
    int a,b,c;
}edge[MAXM];
int f[MAXN];
int cmp(Node a,Node b)
{
    return a.c<b.c;
}
int N,M;
int cnt;
int find(int a)
{
    if(a!=f[a])
        f[a]=find(f[a]);
    return f[a];
}
void init()
{
    for(int i=1;i<=N;i++)
        f[i]=i;
}
void Union(int a,int b)
{
    int fa=find(a);
    int fb=find(b);
    if(fa<fb)
    {
        f[fb]=fa;
        cnt++;
    }
    if(fa>fb)
    {
        f[fa]=fb;
        cnt++;
    }
}
int solve()
{
    sort(edge,edge+M,cmp);

    int ans=INF;
    int max_len,min_len;
    for(int i=0;i<M;i++)
    {init();
        min_len=edge[i].c;
        cnt=0;
        for(int j=i;j<M;j++)
        {
            int a=edge[j].a;
            int b=edge[j].b;
            Union(a,b);
            if(cnt==N-1)
            {
                max_len=edge[j].c;
                //cout<<"max="<<max_len<<endl;
                //cout<<"min="<<min_len<<endl;
                ans=MIN(ans,max_len-min_len);
                //cout<<"ans="<<ans<<endl;
                break;
            }
        }
    }
    if(ans==INF)
        ans=-1;
    printf("%d\n",ans);
    return 0;
}
int main()
{
    while(scanf("%d%d",&N,&M)!=EOF)
    {
        if(N==0&&M==0)
            break;
        for(int i=0;i<M;i++)
            scanf("%d%d%d",&edge[i].a,&edge[i].b,&edge[i].c);
        solve();
    }
    return 0;
}