//给定一个 n 个点 m 条边的无向图,图中可能存在重边和自环,边权可能为负数。
//时间复杂度O(mlogn) 适用于稠密图
#include<bits/stdc++.h>
#define N 1000000
using namespace std;
int n,m,cnt,vis[N],dist[N],head[N],ans;
typedef pair<int,int>PII;
struct edge
{
int to,next,weight;
}e[N];
void add(int u,int v,int w)
{
e[++cnt].to=v;
e[cnt].weight=w;
e[cnt].next=head[u];
head[u]=cnt;
}
void Prim()
{
memset(dist,0x3f,sizeof(dist));
priority_queue<PII,vector<PII>,greater<PII> >heap;
dist[1]=0;
heap.push({0,1});
while(!heap.empty())
{
PII t=heap.top();
heap.pop();
vis[t.second]=1;
for(int i=head[t.second];i;i=e[i].next)
{
if(!vis[e[i].to]&&e[i].weight<dist[e[i].to])
{
dist[e[i].to]=e[i].weight;
heap.push({e[i].weight,e[i].to});
}
}
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);
add(v,u,w);
}
Prim();
for(int i=1;i<=n;i++)
{
if(dist[i]==0x3f3f3f3f)
{
cout<<"impossible";
exit(0);
}
}
for(int i=2;i<=n;i++) ans+=dist[i];
cout<<ans;
return 0;
}
作者:Galaxyq
链接:https://www.acwing.com/activity/content/code/content/1803286/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
最小生成树MSI-Kruskal算法
最新推荐文章于 2024-09-12 19:02:20 发布