Prim算法
介绍
最小生成树
最小生成树是指在一个无向图连通中求得连通所有点的一条路径,且这条路径的所有边的权值之和最小
,此时无向图的这个子图便称为最小生成树
Prim算法
prim算法是图论中用来求最小生成树
的算法,与Dijkstra最短路算法求每个点到起点的最短距离不同,prim算法需要求的是每个点到最小生成树的这个集合
的最小距离
步骤
- 初始化所有点到最小生成树的距离为
∞(0x3f3f3f3f)
- 进行
n
次迭代,每次求出集合外距离最小生成树最短的点,加入最小生成树集合 - 求出距离集合距离最短的点
t
,若它与最小生成树是连通的,则加入最小生成树,否则退出,更新最小生成树权值之和res
- 更新所有点到集合的最短距离
dist[j] = min(dist[j],g[t][j])
时间复杂度 O ( n 2 ) O(n^2) O(n2)
从上面的步骤中我们知道首先要进行n次循环,每次求出一个点离集合的最小值,其次每次循环中我们都需要进行n次循环更新每个点到集合的最短距离,因此时间复杂度为 O ( n 2 ) O(n^2) O(n2)
例题
题目描述
给定一个n
个点m
条边的无向图,图中可能存在重边
和自环
,边权可能为负数
。
求最小生成树的树边权重之和
,如果最小生成树不存在则输出impossible
。
输入格式
第一行包含两个整数n
和m
。
接下来m
行,每行包含三个整数 u,v,w
,表示点u
和点v
之间存在一条权值为w
的边。
输出格式
共一行,若存在最小生成树,则输出一个整数,表示最小生成树的树边权重之和,如果最小生成树不存在则输出impossible
。
数据范围
1
≤
n
≤
500
1≤n≤500
1≤n≤500,
1
≤
m
≤
1
0
5
1≤m≤10^5
1≤m≤105,
图中涉及边的边权的绝对值均不超过
10000
10000
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;
}