成仙之路−> 算法基础课题解
思路:
1. 随便选取一个点加入集合
2. 找到距离集合最近的一个点并加入集合
3. res 加上该点到集合的距离,因为可能存在自环所以要在更新之前执行这个操作
4. 用该点去更新其他点到集合的距离
5. 最后输出答案时要提前存储,防止 prim() 运行两次
6. 稠密图尤其是完全图用 prim 算法效率更佳哟!
完整代码 O(n2+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;
}
/* Prim算法 */ #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() { 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; } } 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); } int t = prim(); if (t == INF) cout << "impossible" << endl; else cout << t << endl; return 0; }