朴素Prim算法求最小生成树
AcWing 858. Prim算法求最小生成树:图解+详细代码注释(带上了保存路径) - AcWing
#include<iostream>
#include<cstring>
using namespace std;
const int N = 510,INF = 0x3f3f3f3f;
int n,m;
int g[N][N];
int dist[N];//这里的dist表示每个点 到生成树的最小距离
bool st[N];
int prim()
{
memset(dist,0x3f,sizeof 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;
//为了防止t点自己有负权边,会自己更新自己
//因此我们先将res进行更新,再用t点去更新其他点的dist
//并且注意,第一个点加进去的时候,这个时候还没有边,千万不要更新res
//不然res会在第一个点的时候就更新成INF了
if(i) res += dist[t];
st[t] = true;
//更新其他点的dist
for(int j = 1;j<=n;j++) dist[j] = min(dist[j],g[t][j]);
}
return res;
}
int main()
{
cin>>n>>m;
memset(g,0x3f,sizeof g);
while(m--)
{
int a,b,c;
cin>>a>>b>>c;
g[a][b] = g[b][a] = min(g[a][b],c);
}
int t = prim();//返回值为最小生成树中所有边权数之和
if(t == INF)puts("impossible");
else cout<<t<<endl;
return 0;
}