AcWing 858. Prim算法求最小生成树
原题链接
简单
作者:
wangyj
,
2020-10-20 19:46:31
,
所有人可见
,
阅读 255
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int BIG=0x3f3f3f3f;
int n,m,g[510][510],dist[510];
bool st[510];
int prim()
{
memset(dist,0x3f,sizeof(dist));
int res=0,i,t,j;
for(i=0;i<n;i++){
t=-1;
for(j=1;j<=n;j++)if(!st[j]&&(t==-1||dist[t]>dist[j]))t=j;
if(i&&dist[t]==BIG)return BIG;
if(i)res+=dist[t];
for(j=1;j<=n;j++)dist[j]=min(dist[j],g[t][j]);
st[t]=true;
}
return res;
}
int main()
{
int a,b,c,t;
scanf("%d%d",&n,&m);
memset(g,0x3f,sizeof(g));
while(m--){
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=min(g[a][b],c);
}
t=prim();
if(t==BIG)printf("impossible\n");
else printf("%d\n",t);
}
我已经尽力了,大家如果有什么改进意见,一定要告诉我,谢谢!