Acwing-基础算法课笔记之搜索与图论(Prim算法)
一、Prim算法
1、概述
Prim算法的思想和Dijkstra算法基本相同,只有一个区别:Dijkstra算法需要更新集合中的所有点到起点的距离,而Prim算法不需要。和Dijkstra算法一样,Prim算法编码时如果用优先队列查找集合中最近的点,能优先算法,此时总时间复杂度为 O ( m log 2 n ) O(m\log_{2}n) O(mlog2n)。
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 ) O(n^2+m) O(n2+m), n n n表示点数, m m 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;
}