Prim模板

通过代码探索Prim算法：最小生成树之旅

在计算机科学领域，图算法占据了至关重要的位置，尤其是在设计高效的网络（无论是社交网络、计算机网络还是交通网）时。在这些算法中，寻找最小生成树（MST）是一个基本问题，它寻求以最小的总边权重连接图中所有顶点（无任何循环）所需的最小边集。Prim算法作为解决此问题的经典且高效的方法脱颖而出。让我们通过详细检查代码实现及算法的基本原理来深入探究它的复杂之处。

理解Prim算法

Prim算法是一种贪心方法，它一次添加一个顶点来构建MST，从任意顶点开始。它维护了两个顶点集合：已经包含在MST中的和尚未包含的。在每一步中，它考虑连接这两个集合的所有边，并从这些边中选择最小权重的边。然后，它将此边的另一端的顶点移动到包含在MST中的顶点集合里。

Prim算法的美在于它的简单和高效。它通过添加最接近树的顶点来迭代扩展MST，直到包含所有顶点。

代码解析

提供的代码片段是Prim算法在C++中的完整实现。让我们来详细分析它的主要组成部分：

初始设置

#include<bits/stdc++.h>
using namespace std;

const int N = 510, INF = 0x3f3f3f3f;
int g[N][N], dist[N];
bool st[N];
int n, m;

代码以包含所有标准库开始，并定义了常量和全局变量。N是图可能拥有的顶点的最大数量，INF代表无限距离，用于初始化。图g被表示为邻接矩阵，dist用于跟踪每个顶点到MST的最小距离，st跟踪顶点是否已包含在MST中。

图的构建和初始化

memset(g, 0x3f, sizeof g);
cin >> n >> m;
for(int i = 1; i <= m; i++) {
    int a, b, c;
    cin >> a >> b >> c;
    g[a][b] = g[b][a] = min(g[a][b], c);
}

在这一部分中，代码首先使用一个大数（0x3f3f3f3f）初始化邻接矩阵g，这意味着初始时，任意两个顶点之间没有直接的连接。然后，它读取顶点数n和边数m，并根据输入的每条边更新邻接矩阵。如果两个顶点之间有多条边，则保留权重最小的那条边。

Prim算法的实现

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];
        for(int j = 1; j <= n; j++) dist[j] = min(dist[j], g[t][j]);
        st[t] = true;
    }
    return res;
}

prim函数是算法的核心，它首先使用0x3f3f3f3f初始化dist数组。然后，它通过一个循环，每次迭代选择一个与MST的距离最短的未包含顶点t，并将其加入MST。如果在任何时刻t的dist[t]为INF，这意味着图不连通，函数返回INF。否则，它更新结果res并调整dist数组，以反映新加入的顶点对其他所有顶点距离的影响。

主函数和结果输出

int main() {
    int t = prim();
    if(t == INF) puts("impossible");
    else cout << t << endl;
    return 0;
}

主函数调用prim函数来计算MST的总权重，并根据返回值输出结果。如果图不连通，它输出"impossible"；否则，输出MST的总权重。

代码

#include<bits/stdc++.h>

using namespace std;

const int N = 510, INF = 0x3f3f3f3f;
int g[N][N], dist[N];
bool st[N];
int n, m;

int prim()
{
    memset(dist, 0x3f, sizeof dist);

    int res = 0;
    for(int i = 0; i < n; i++)//第一个i从0开始是为了后面if语句中只处理n - 1条边
    {//i=0时相当于预处理dist数组，使得dist数组中存在数字，使其能在下一个循环中能找出离生成树最近的一个点，并以此来更新边的距离
        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];//先更新防止负的自环

        for(int j = 1; j <= n; j++) dist[j] = min(dist[j], g[t][j]);

        st[t] = true; 
    }

    return res;
}

int main()
{
    memset(g, 0x3f, sizeof g);
    cin >> n >> m;
    for(int i = 1; i <= m; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        g[a][b] = g[b][a] = min(g[a][b], c);
    }

    int t = prim();

    if(t == INF) puts("impossible");
    else cout << t << endl;

    return 0;
}

总结

通过这段代码和对Prim算法的探讨，我们不仅加深了对这一经典算法的理解，还体会到了算法在解决实际问题中的应用。Prim算法以其简洁和高效，在图算法中占有一席之地，是理解和掌握图论基础的关键步骤。希望这篇博客能够帮助你在探索图算法的旅程中迈出坚实的一步。