题目描述
最小生成树问题是实际生产生活中十分重要的一类问题。假设需要在n个城市之间建立通信联络网,则连通n个城市只需要n-1条线路。这时,自然需要考虑这样一个问题,即如何在最节省经费的前提下建立这个通信网。
可以用连通网来表示n个城市以及n个城市之间可能设置的通信线路,其中网的顶点表示城市,边表示两个城市之间的线路,赋于边的权值表示相应的代价。对于n个顶点的连通网可以建立许多不同的生成树,每一棵生成树都可以是一个通信网。现在,需要选择一棵生成树,使总的耗费最小。这个问题就是构造连通网的最小代价生成树,简称最小生成树。一棵生成树的代价就是树上各边的代价之和。
而在常用的最小生成树构造算法中,普里姆(Prim)算法是一种非常常用的算法。以下是其算法的大致结构:
在本题中,读入一个无向图的邻接矩阵(即数组表示),建立无向图并按照以上描述中的算法建立最小生成树,并输出最小生成树的代价。
输入
输入的第一行包含一个正整数n,表示图中共有n个顶点。其中n不超过50。
以后的n行中每行有n个用空格隔开的整数,对于第i行的第j个整数,如果不为0,则表示第i个顶点和第j个顶点有直接连接且代价为相应的值,0表示没有直接连接。当i和j相等的时候,保证对应的整数为0。
输入保证邻接矩阵为对称矩阵,即输入的图一定是无向图,且保证图中只有一个连通分量。
输出
只有一个整数,即最小生成树的总代价。请注意行尾输出换行。
样例输入
4
0 2 4 0
2 0 3 5
4 3 0 1
0 5 1 0
样例输出
6
提示
在本题中,需要掌握图的深度优先遍历的方法,并需要掌握无向图的连通性问题的本质。通过求出无向图的连通分量和对应的生成树,应该能够对图的连通性建立更加直观和清晰的概念。
代码:
#include <stdio.h>
int n,min;
int book[60];
int e[60][60];
int dis[60];
int inf=99999999;
int main()
{
int i,j,k;
int count,sum;
scanf("%d",&n);
for(i=0;i<n;i++)
for(j=0;j<n;j++)
scanf("%d",&e[i][j]);
for(i=0;i<n;i++)
for(j=0;j<n;j++)
if(e[i][j]==0)
e[i][j]=inf;
count=0;sum=0;
//初始化dis数组,这里是0号顶点到各个顶点的初始距离
for(i=0;i<n;i++)
dis[i]=e[0][i];
book[0]=1;
count++;
while(count<n)
{
min=inf;
for(i=0;i<n;i++)
if(book[i]==0&&dis[i]<min){
min=dis[i];
j=i;
}
book[j]=1;
count++;
sum+=dis[j];
for(k=0;k<n;k++){
if(book[k]==0&&dis[k]>e[j][k])
dis[k]=e[j][k];
}
}
/*for(i=0;i<k;i++)
printf("%d ",dis[i]);*/
printf("%d\n",sum);
return 0;
}
Prime算法:将图中所有的顶点分为两类:树顶点(已被选入生成树的顶点)和非树顶点(未被选入生成树的顶点)。首先选择任意一个顶点加入生成树(可以理解为生成树的根)。接下来要找出一条边添加到生成树,这需要枚举每一个树顶点到每一个非树顶点所有的边,然后找到最短边加入到生成树。照此方法,重复操作n-1次,直到所有的顶点都加入生成树中。
对于本题,整理了一下算法流程:
1:从任意一个顶点开始构造生成树,假设就从0号顶点吧,首先将顶点0号加入生成树中,用一个一维数组book来标记那些顶点已经加入了生成树。
2:用数组dis记录生成树到各个顶点的距离。最初生成树中只有0号顶点,有直连边时,数组dis中存储的就是0号顶点到该顶点的边的权值,没有直连边的时候就是无穷大,即初始化dis数组。
3:从数组dis中选出离生成树最近的顶点(假设这个顶点为j)加入到生成树中(即在数组dis中找最小值)。再以j为中间点,更新生成树到每一个非树顶点的距离(就是松弛),即如果dis[k]>e[j][k]则更新dis[k]=e[j][k]。
4:重复第三步,直到生成树中有n个顶点为止。