C++数据结构与算法分析——Prim算法

Prim算法

介绍

最小生成树

最小生成树是指在一个无向图连通中求得连通所有点的一条路径，且这条路径的所有边的权值之和最小，此时无向图的这个子图便称为最小生成树

Prim算法

prim算法是图论中用来求最小生成树的算法,与Dijkstra最短路算法求每个点到起点的最短距离不同，prim算法需要求的是每个点到最小生成树的这个集合的最小距离

步骤

1. 初始化所有点到最小生成树的距离为∞(0x3f3f3f3f)
2. 进行n次迭代，每次求出集合外距离最小生成树最短的点，加入最小生成树集合
3. 求出距离集合距离最短的点t，若它与最小生成树是连通的，则加入最小生成树，否则退出，更新最小生成树权值之和res
4. 更新所有点到集合的最短距离dist[j] = min(dist[j],g[t][j])

时间复杂度$O(n^2)$

从上面的步骤中我们知道首先要进行n次循环，每次求出一个点离集合的最小值，其次每次循环中我们都需要进行n次循环更新每个点到集合的最短距离，因此时间复杂度为$O(n^2)$

例题

题目描述

给定一个n个点m条边的无向图，图中可能存在重边和自环，边权可能为负数。
求最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

输入格式

第一行包含两个整数n和m。
接下来m行，每行包含三个整数 u，v，w，表示点u和点v之间存在一条权值为w的边。

输出格式

共一行，若存在最小生成树，则输出一个整数，表示最小生成树的树边权重之和，如果最小生成树不存在则输出impossible。

数据范围

$1\le n\le 500$,
$1\le m\le 10^5$,
图中涉及边的边权的绝对值均不超过$10000$。

输入样例：

4 5
1 2 1
1 3 2
1 4 3
2 3 2
3 4 4

输出样例：

6

解题思路

因本题是稠密图，因此我们需要用邻接矩阵来存图。这题是一个最小生成树问题，可以使用Prim算法解答，需要解决重边和自环问题，首先自环问题只在最后一步更新所有最短路时才可能出现，因此只要我们先更新res将点t加入集合中后，t将不再参与最小生成树的构建，因此答案不会计入自环距离，重边问题只需要在最开始初始化g[t][j]时将最短的g[t][j]存下来即可。

代码

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;

const int N = 510,INF = 0x3f3f3f3f;
int g[N][N]; // 邻接矩阵,g[i][j]表示点i到点j的距离
int dist[N]; // dist[i]表示点i到集合的距离
bool st[N]; // 存储已加入连通块的点(已加入集合的点)
int n,m;

int prim(){ // prim算法
    int res = 0; // 存最小生成树的树边权重之和
    memset(dist,0x3f,sizeof dist); // 把所有点离集合的距离初始化为正无穷
    
    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])) // 假如(j在集合外) && (t未初始化 || j离集合距离小于t离集合距离)则更新t = j
                t = j;
        
        if(i && dist[t] == INF) return INF; // 假如图不连通则表示无法生成最小生成树
        if(i) res += dist[t]; // 假如i不是第一个点，则将该点到集合的距离加入res
        st[t] = true; // 将点t加入集合
        
        for(int j = 1; j <= n; j ++) dist[j] = min(dist[j],g[t][j]); // 更新集合外所有点与最小生成树的最短距离
    }
    
    return res;
}

int main(){
    scanf("%d%d",&n,&m);
    memset(g,0x3f,sizeof g); // 先把所有点距离初始化为正无穷
    while(m --){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        g[a][b] = g[b][a] = min(g[a][b],c); // 因为是无向图，所以两个点互相连通
    }
    
    int t = prim();
    if(t == INF) puts("impossible");
    else printf("%d",t);
    
    return 0;
}