【C++简单算法】Prim与最小生成树

  对于给定的一个无向图与权值，如何求他的最小生成树呢？一个比较经典的做法是Prim算法。
  主要思路是，首先定义个邻接矩阵，里面保存着每条无向边的权值，如果两个点间没有通路，则定义为INT_MAX。再定义个最小边矩阵，保存未连接的点跟已连接点的最短距离。还要定义一个矩阵，用来标识点是否已经被加入到生成树当中。
  接着决定一个点作为起点，并把最小边矩阵更新为每个点到选定点的距离。然后遍历一遍，找出最小边矩阵中最小的边，将那个所连接的点加入到生成树当中，将这个点记为P。同时更新最小边矩阵，更新方法为：
  对于每一个不在生成树的点，如果它到P的距离小于最小边矩阵的值，就更新。否则保持原来的值。

  举个例子，有以下无向图，将1作为起点

  则可以得到最小边矩阵

  可以看到最小的是3这个边，将3加入最小生成树中，并更新矩阵。

  此时最小的是连接4的这个边，将4加入树中，继续更新矩阵。

  最后把2加入树中，就得到了最小生成树。

  C++代码如下：

#include<iostream>
using namespace std;
//假设输入的点个数在500以内
int dis[510][510];
int vis[510];
int min_dis[510];
int main()
{
    //先输入点的个数和边的个数
    int n, m;
    int ans = 0;
    
    cin >> n >> m;
    for (int i = 0; i <= n; i++) {
        for (int j = 0; j <= n; j++) {
            dis[i][j] = INT_MAX;
        }
    }
    while (m--) {
        int u, v, w;
        //无向图，u到v的权值是w
        cin >> u >> v >> w;
        dis[u][v] = w;
        dis[v][u] = w;
    }
    //首先把1加入生成树中
    vis[1] = 1;
    //第一次更新最小边矩阵
    for (int i = 2; i <= n; i++) {
        min_dis[i] = dis[1][i];
    }
    //用来标识是否找到一个点，防止因为所有点不在一个图内而多做动作
    bool flag = true;
    //for (int i = 0; i <= n; i++)cout << dis[1][i] << endl;
    for (int i = 2; i <= n; i++) {
        int min_now=INT_MAX;
        int now=1;
        flag = false;
        for (int i = 2;i<=n; i++) {
            if (min_dis[i] < min_now && vis[i]==0 ) {
                min_now = min_dis[i];
                now = i;
                flag = true;
            }

        }
        if (flag) {
            vis[now] = 1;
            ans += min_now;
            //更新最小边矩阵
            for (int i = 2; i <= n; i++) {
                if (min_dis[i] > dis[now][i] && vis[i] == 0) {
                    min_dis[i] = dis[now][i];
                }
            }

        }
        else { break; }
    }
    for (int i = 2; i <= n; i++) {
        if (vis[i] == 0) { 
            //无法形成一个最小生成树的情况
            cout << "impossible" << endl; 
            return 0;
        }
    }
    cout << ans << endl;
    return 0;
}