$\huge \color{orange}{成仙之路->}$ $\huge \color{purple}{算法基础课题解}$
思路:
1. 随便选取一个点加入集合
2. 找到距离集合最近的一个点并加入集合
3. res 加上该点到集合的距离,因为可能存在自环所以要在更新之前执行这个操作
4. 用该点去更新其他点到集合的距离
5. 最后输出答案时要提前存储,防止 prim() 运行两次
6. 稠密图尤其是完全图用 prim 算法效率更佳哟!
完整代码 $O(n^2+m)$
#include<bits/stdc++.h>
using namespace std;
const int N = 510, INF = 0x3f3f3f3f;
int n,m;
int g[N][N],dist[N];
bool st[N];
int prim()
{
//初始化dist数组
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[j]<dist[t]))
t=j;
//说明点不连通,直接返回INF
if(i&&dist[t]==INF) return 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()
{
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);
}
//用t提前存储,防止prim()运行两次
int t=prim();
if(t==INF) cout<<"impossible"<<endl;
else cout<<t<<endl;
return 0;
}