Kruskal算法求最小生成树模板

最新推荐文章于 2024-10-02 22:45:05 发布

布克波波

最新推荐文章于 2024-10-02 22:45:05 发布

阅读量93

点赞数

文章标签：算法 c++ 图搜索

本文链接：https://blog.csdn.net/Bookerbobo/article/details/123745953

版权

acwing859https://www.acwing.com/problem/content/861/

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2e5+10;
int p[N];
struct node
{
    int u,v,w;
}e[N];

int find(int x)//压缩路径的并查集find函数
{
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}
bool cmp(node x,node y)
{
    return x.w<y.w;
}

int main()
{
    int n,m,a,b,c,res=0,cnt=0;
    cin>>n>>m;
    for(int i=0;i<m;i++)
    {
        cin>>a>>b>>c;
        e[i]={a,b,c};
    }
   for(int i=1;i<=n;i++)p[i]=i;
   sort(e,e+m,cmp);
   for(int i=0;i<m;i++)
   {
       a=e[i].u,b=e[i].v,c=e[i].w;
       a=find(a),b=find(b);
       if(a!=b).//使用并查集检查新加边是否成环
       {
           res+=c;
           p[a]=b;
           cnt++;
       }
   }
   if(cnt<n-1)cout<<"impossible";
   else cout<<res;
    return 0;
}