朴素Prim算法

算法简介

普里姆算法（Prim算法），是一种求最小生成树的方法，时间复杂度为O(n^2)

算法流程

朴素Prim求解最小生成树的方法与Dijkstra算法很相似，可以参考学习。

朴素Prim算法通过n次遍历来确定n个点，首先将寻找到的距离集合最近的点，将这个点加入集合，再拿这个点来更新不在集合中的点离集合距离。

每次遍历是先找点，再更新其他点的距离，这与Dijkstra算法相同。

实例

模板题

acwing858. Prim算法求最小生成树
给定一个n个点m条边的无向图，图中可能存在重边和自环，边权可能为负数。

求最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

给定一张边带权的无向图G=(V, E)，其中V表示图中点的集合，E表示图中边的集合，n=|V|，m=|E|。

由V中的全部n个顶点和E中n-1条边构成的无向连通子图被称为G的一棵生成树，其中边的权值之和最小的生成树被称为无向图G的最小生成树。

输入格式

第一行包含两个整数n和m。
接下来m行，每行包含三个整数u，v，w，表示点u和点v之间存在一条权值为w的边。

输出格式

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

数据范围

1≤n≤500,

1≤m≤1e5,

图中涉及边的边权的绝对值均不超过10000。

输入样例：

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4

输出样例：

6

分析

这是一道Prim的模板题，需要注意的的是图中有重边和自环的存在，一般的稠密图使用朴素Prim来求解。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=510;
const int INF=0x3f3f3f3f;
int g[N][N];
int d[N];
bool f[N];
int n,m;
int prim()
{
    memset(d,0x3f,sizeof d);
    int res=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=1;j<=n;j++)
            if(!f[j]&&(t==-1||d[t]>d[j]))
                t=j;
        if(i&&d[t]==INF) return INF;
        if(i) res+=d[t];//res增加不能放在跟新点之后，因为图中有自环的存在，更新后d[t]可能变换。
        f[t]=1;
        for(int j=1;j<=n;j++)
            if(!f[j])
                d[j]=min(d[j],g[t][j]);
        
    }
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g);
    for(int i=0;i<m;i++)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        g[a][b]=g[b][a]=min(g[a][b],w);
    }
    int k=prim();
    if(k==INF) printf("impossible\n");
    else printf("%d\n",k);
    return 0;
}

另外，堆优化版的Prim算法和堆优化的Dijkstra算法也很相似，时间复杂度也相同。所以一般在求解稀疏图时，不用堆优化的Prim,而是用Kruskal算法，Kruskal算法的时间复杂度为$O(mlogm)$与$O(mlogn)$差不多，所以在求解稀疏图时用Kruskal算法。