最小生成树

题目描述

最小生成树问题是实际生产生活中十分重要的一类问题。假设需要在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个顶点为止。

 

 

[提交][状态]