对于给定的一个无向图与权值,如何求他的最小生成树呢?一个比较经典的做法是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;
}