AcWing 859. Kruskal算法求最小生成树
原题链接
简单
作者:
xiaoqi_
,
2021-04-01 10:22:35
,
所有人可见
,
阅读 369
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100010,M=200010,INF=0x3f3f3f3f;
int n,m;
int p[N];
struct Edge
{
int a,b,w;
bool operator<(const Edge &W) const{ //重载运算符
return w<W.w; //按w进行排序
}
}edges[M];
int find(int x) //并查集模板
{
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int kruskal()
{
sort(edges,edges+m);
for(int i=1;i<=n;i++) p[i]=i; //并查集初始化
int res=0,cnt=0; //路径总和 边的数量
for(int i=0;i<m;i++)
{
int a=edges[i].a,b=edges[i].b,w=edges[i].w;
a=find(a),b=find(b);
if(a!=b) //如果两个不在同一集合,那么将两个集合合并
{
p[a]=b; //加入集合
res+=w; //路径+权值
cnt++; //边数++
}
}
if(cnt<n-1) return INF; //最后如果总边数小于点数-1,那么说明不能把所有的点加入到集合中。
return res;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=0;i<m;i++)
{
int a,b,w;
scanf("%d%d%d",&a,&b,&w);
edges[i]={a,b,w};
}
int t=kruskal();
if(t==INF) puts("impossible");
else printf("%d\n",t);
return 0;
}