简述
prim和Kruskal都是解决最小生成树的算法,Kruskal主要用于解决顶点多但边稀疏的图,prim主要用于解决边稠密的图。
c++
ll prim(int n) {
memset(vis, 0, sizeof(vis));
for(int i = 1; i <= n; i++) {
d[i] = e[1][i];
}
vis[1] = 1;
ll ans = 0;
for(int k = 2; k <= n; k++) {
int x, m = INF;
for(int i = 1; i <= n; i++) {
if(!vis[i] && d[i]<m) {
x = i;
m = d[i];
}
}
vis[x] = 1;
ans += m;
for(int i = 1; i <= n; i++) {
if(!vis[i] && e[x][i]<d[i])
d[i] = e[x][i];
}
}
return ans;
}