AcWing 3210-csp-kruskal算法求最小生成树. 最优灌溉
原题链接
简单
作者:
YAX_AC
,
2024-11-14 18:53:46
,
所有人可见
,
阅读 2
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N = 100010;
int p[N];
int n,m;
struct E{
int a,b,w;
}edg[N*2];
bool cmp(E a,E b)
{
return a.w<b.w;
}
int find(int a)
{
if(p[a]!=a) p[a] = find(p[a]);
return p[a];
}
int main()
{
cin>>n>>m;
for(int i = 1; i<=n; i++) p[i] = i;//初始化并查集
for(int i = 1; i<=m; i++)
{
int a,b,c;
cin>>a>>b>>c;
edg[i] = {a,b,c};
}
sort(edg+1,edg+m+1,cmp);
int res = 0,cnt = 0;//res是权值,cnt是边数
//kruskal算法:从最小的边开始化,直到把整个图画通
for(int i = 1; i<=m; i++)
{
int a = edg[i].a,b = edg[i].b,w = edg[i].w;
a = find(a),b = find(b);
if(a!=b)//如果不连通
{
p[a] = b;//两个集合合并
res += w;
cnt++;//当前加了多少条边
}
}
//如果保留的边小于点数n-1,则不能连通
if(cnt<n-1) puts("impossible");
else cout<<res;
return 0;
}