AcWing 859. Kruskal算法求最小生成树
原题链接
简单
作者:
tzp2333
,
2020-08-23 22:28:57
,
所有人可见
,
阅读 445
// 使用优先队列 (其实就是堆排) 堆中的数据的类型 为 pair<int,pair<int,int>>
#include <iostream>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200010;
int p[N],res;
int n,m;
typedef pair<int,pair<int,int>> PIII;
priority_queue<PIII,vector<PIII>,greater<PIII>> q;
int find (int x)
{
if (p[x]!=x) p[x]=find(p[x]);
return p[x];
}
int main ()
{
cin>>n>>m;
while (m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
q.push({c,{a,b}});
}
for (int i = 1;i<=n;i++) p[i]=i;
int k =0;
while (k<=n-1&&q.size())
{
PIII t = q.top();
q.pop();
int c=t.first,a=t.second.first,b=t.second.second;
int x=find(a),y=find(b);
if (x!=y)
{
p[x]=y;
k++;
res += c;
}
}
if (k<n-1) printf("impossible");
else printf("%d",res);
}