kruskal算法模板

SAVAv

于 2024-07-10 19:20:26 发布

阅读量384

点赞数 6

文章标签：算法

本文链接：https://blog.csdn.net/hui_le4/article/details/140332304

版权

#include<bits/stdc++.h>
using namespace std; 
const int N=2e5+10;

int n,m,k;
int p[N];
struct Edge{
	int a,b,w;
   bool	operator <(const Edge &A) const//重载运算符
   {
   	return w<A.w;
   }
}edge[N];


int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}

int  kruskal()//
{
	int res=0,cnt=0;//
	for(int i=0;i<m;i++)
	{
		int a=edge[i].a,b=edge[i].b,w=edge[i].w;
		if(find(a)!=find(b))//如果根节点不同
		{
			p[find(a)]=p[find(b)];
			cnt++;
			res+=w;
		}
	}
	if(cnt==n-1) return res;
	else return 0x3f3f3f3f;
	
}   
    


signed main()
{ 
 
 cin>>n>>m;
 for(int i=1;i<=n;i++)
 p[i]=i;
 for(int i=0;i<m;i++)
 {
 	int a,b,c;
 	cin>>a>>b>>c;
 	edge[i]={a,b,c};
 }
 sort(edge,edge+m);//排序
 int ans=kruskal();
 if(ans==0x3f3f3f3f) cout<<"impossible";
 else cout<<ans;
 
 
 
 
 
}

859. Kruskal算法求最小生成树 - AcWing题库