AcWing 858. Prim算法求最小生成树
原题链接
简单
作者:
TaoZex
,
2019-08-05 14:17:13
,
所有人可见
,
阅读 1013
#include<bits/stdc++.h>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
int g[N][N];
int d[N];
bool st[N];
int n,m;
int prim(){
memset(d,INF,sizeof(d));
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||d[j]<d[t])){ //找出不在集合中离集合最近的点
t=j;
}
}
if(i&&d[t]==INF) return INF; //判断是否连通,具有最小生成树
//(有些图有负权边,最小生成树权值和恰好为-1,返回-1就会出错)
if(i) res+=d[t];
st[t]=true; //将t加入集合,并更新最小生成树权值
for(int j=1;j<=n;j++) d[j]=min(d[j],g[t][j]); //更新所连点到集合的最短距离
}
return res;
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
if(i==j) g[i][j]=0;
else g[i][j]=INF;
}
}
while(m--){
int a,b,c;
cin>>a>>b>>c;
g[a][b]=g[b][a]=min(g[a][b],c);
}
int ans=prim();
if(ans==INF) puts("impossible");
else cout<<ans<<endl;
}