prim算法的精髓不在于点的距离更新,这样永远把握不住要领,而在于生成树的更新,每次都要把剩余点和生成树的最小距离重新求出来。
所以我们首先会需要一个访问数组,表示当前点是否在生成树里面,一方面是因为既然在生成树里面了,我们就不需要考虑了。另一方面也可以认为这个距离已经变成0了。
在加入节点后,如何更新其余节点到我们的生成树的最小距离?策略很重要,当树里面只有初始节点的时候,这个最小距离就是到这个初始点的距离。加入了新的节点之后,这里不一定会导致原来的最短距离失效,只是,我们必须重新枚举,这样相当于每次最后一个点都是前面所有点斗争比较之后的结果。
在这个过程中,选择一个点重新加入这棵最小生成树,所以需要一个标记t,来表示我们选择了哪一个点,这个点也是遍历所有剩余点选出的最小距离的点,同时也可以利用这个计算最后的最小生成树权值。
其实,隐含的,我们发现我们也需要一个距离数组。
你看,我们就这样从无推出了算法。最后说明一下,prim核心要点是点贪心。Krusal核心是边贪心。
int dist[N]; // 存储其他点到当前最小生成树的距离
bool st[N]; // 存储每个点是否已经在`在这里插入代码片`生成树中
int prim()
{
for (int i = 0; i < N; i++) {
dist[i] = INF;
}
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;
}