联系:Dijkstra算法是更新到起始点的距离,Prim是更新到集合S的距离
#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()
{
int res = 0;
memset(dist, 0x3f, sizeof dist);
for (int i = 0; i < n ; i ++ )
{
int t = -1;
// 1.找集合之外距离最近的点
for (int j = 1; j <= n ; j ++ )
{
if (!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
}
// 2. 标记这个点(并加入集合)
st[t] = true;
if (i && dist[t] == INF) return INF; //如果不是第一个点还距离∞,说明图不连通
if (i) res += dist[t]; // 不是第一个点,加入集合
// 3. 用t更新其他点到集合的距离
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;
}