Acwing-基础算法课笔记之搜索与图论（Prim算法）

一、Prim算法

1、概述

Prim算法的思想和Dijkstra算法基本相同，只有一个区别：Dijkstra算法需要更新集合中的所有点到起点的距离，而Prim算法不需要。和Dijkstra算法一样，Prim算法编码时如果用优先队列查找集合中最近的点，能优先算法，此时总时间复杂度为$O(m{\log }_{2}n)$。

2、过程模拟

设最小生成树中的点的集合为U，开始时最小生成树为空，所以U为空。
如图所示：

（1）任取一点，如点1，放到U中，U={1}；
（2）找离集合U中的点最近的邻居，即点1的邻居，是点2，放到U中，U={1，2}；
（3）找离U最近的点，是点5，U={1，2，5}；
（4）与U距离最短的是点1和点5之间的边，但是它没扩展新的点，不符合要求；
（5）加入点4，U={1，2，5，4}；
（6）加入点3，U={1，2，5，4，3}，所有点都在U中，结束。

3、Prim算法模板

时间复杂度是$O(n^2+m)$，$n$表示点数，$m$表示边数

int n;      // n表示点数
int g[N][N];        // 邻接矩阵，存储所有边
int dist[N];        // 存储其他点到当前最小生成树的距离
bool st[N];     // 存储每个点是否已经在生成树中
// 如果图不连通，则返回INF(值是0x3f3f3f3f), 否则返回最小生成树的树边权重之和
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[t] > dist[j]))
                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;
}

Acwing-Prim算法模板