Prim算法适用于稠密图,关于其详解,可参考博客:
https://blog.csdn.net/yeruby/article/details/38615045
代码如下:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int map[105][105],dis[105]; //dis[]记录当前连通块与其它未连接的节点的最小距离
bool vis[105]; //vis表示哪个节点已经加入连通块了,相当于用数组来表示一个集合
int Prim(int n)
{
int ans=0;
memset(vis,0,sizeof(vis));
for(int i=1;i<=n;i++)
dis[i]=map[1][i]; //任选某一节点作为起点,在这里选1
vis[1]=1; //1加入集合
for(int i=1;i<=n;i++) //注意,这里的i并不是表示节点编号,只是一个循环的标记,没有什么太多的意义,可以不理它
{
int temp=1005,k; //temp用来探测dis[]中最小的距离,而k记录最小距离对应的节点编号
for(int j=1;j<=n;j++) //遍历n个节点
if(!vis[j]&&temp>dis[j]) //前提:vis[j]=0,即这个节点还未被加入连通块
{
temp=dis[j];
k=j;
}
if(temp==1005) //如果temp不变,说明所以节点都连通了,就可以结束了
break;
ans+=dis[k];
vis[k]=1; //将k加入集合
for(int j=1;j<=n;j++) //要更新dis[]
if(!vis[j]&&dis[j]>map[k][j])
dis[j]=map[k][j];
}
return ans;
}
int main()
{
int n,q;
while(cin>>n)
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cin>>map[i][j];
cin>>q;
while(q--)
{
int a,b;
cin>>a>>b;
map[a][b]=map[b][a]=0; //一开始,我还在后面添加vis[a]=vis[b]=1,因为当时认为既然他们已经连通了,就应该把a,b加入集合
//这么理解好像没问题,但是因为算法编程实现的原因,这样处理最终会出错,此时我们应将map[a][b]=map[b][a]=0理解为,a与b的距离为0,或连通代价为0
}
cout<<Prim(n)<<endl;
}
return 0;
}