AcWing 858. Prim算法求最小生成树
原题链接
简单
作者:
xiaoqi_
,
2021-03-31 18:25:08
,
所有人可见
,
阅读 201
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N =510,INF=0x3f3f3f3f;
int n,m;
int g[N][N];
int dist[N];
bool st[N];
int prim()
{
memset(dist,0x3f,sizeof dist); //初始化dist距离
int res=0; //树路径的和
for(int i=0;i<n;i++) //迭代每一个点
{
int t=-1;
for(int j=1;j<=n;j++)
if(!st[j]&&(t==-1||dist[t]>dist[j]))
//如果当前点没有在集合中,并且找到了距离集合最近的点
t=j;
if(i&&dist[t]==INF) return INF;
//如果i不是0,并且找到的点到集合的最短距离为INF
//说明集合中的点不能全部相互连接,无生成树
//判断是否连通,有无最小生成树
if(i) res+=dist[t]; //计算距离
st[t]=true; //将这个点加入集合
for(int j=1;j<=n;j++) dist[j]=min(dist[j],g[t][j]); //更新每一个点到当前这个集合的距离
}
return res; //返回路径和
}
int main()
{
scanf("%d%d",&n,&m);
memset(g,0x3f,sizeof g);
while(m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=min(g[a][b],c);//去除重边的写法
}
int t=prim();
if(t==INF) puts("impossible");
else printf("%d\n",t);
return 0;
}