算法学习笔记四:最小生成树(prim)

题目描述
市政府“惠民工程”的目标是在全市n个居民点间之架设煤气管道（但不一定有直接的管道相连，只要能间接通过管道可达即可）。很显然最多可架设 n(n-1)/2条管道,然而实际上要连通n个居民点只需架设n-1条管道就可以了。现请你编写程序，计算出该惠民工程需要的最低成本。
输入
测试输入包含若干测试用例。每个测试用例的第1行给出居民点数目M ( < =100 )、 评估的管道条数 N；随后的 N 行对应居民点间管道的成本，每行给出一对正整数，分别是两个居民点的编号，以及此两居民点间管道的成本（也是正整数）。为简单起见，居民点从1到M编号。
输出
对每个测试用例，在1行里输出全市管道畅通所需要的最低成本。若统计数据不足以保证畅通，则输出“?”。
样例输入
3 3
1 2 1
1 3 2
2 3 4
3 1
2 3 2
样例输出
3
?

//即求最小生成树问题，此处使用prim by邻接矩阵 
#include<iostream>
#include<algorithm> 
#include<memory.h>
using namespace std;
#define INF 100000000//INF为一个很大的数
#define max 110
//const int INF=10000000;
bool vis[max];//记录是否被访问 
int d[max];// 存放顶点v与集合S之间的最短距离 
int map[max][max]; 
int m,n;//按题目要求m即为顶点数，n为边数 
int prim()
{
	int i,j;
	fill(d,d+max,INF);
	memset(vis,false,sizeof(vis));
	d[1]=0;//只有0号顶点到集合S的距离为0，其余为INF 
	int ans=0;//存放最小生成数的边权之和
	for(i=1;i<=m;i++)
	{
		int u=-1,min=INF;//u记录每次找到的距离s最小的顶点， min存放最小距离 
		for(j=1;j<=m;j++)//此循环找到未访问的顶点中d[]最小的点且从1开始 
		{
			if(vis[j]==false&&d[j]<min)//找出d[]最小的顶点放在u 
			{
				u=j;
				min=d[j];
			 } 
		 } 
		 if(u==-1)
		 	return -1;//若找不到小于INF的d[u],说明剩下的顶点和集合S不连通
		 vis[u]=true;
		 ans+=d[u];
		 for(int v=1;v<=m;v++)//此循环目的即：在u加入集合s后更新d[] 
		 {	//v未访问&& u与v间有边&& 以u为中间点可以使v离集合s更近 
		 	if(vis[v]==false&&map[u][v]!=INF&&map[u][v]<d[v])
		 		d[v]=map[u][v];//更新d[v] 
		  } 
	 } 
	 return ans; 
 } 
int main()
{
	while(cin>>m>>n)
	{
		int i,x,y,temp;
		fill(map[0],map[0]+max*max,INF);
		for(i=0;i<n;i++)
		{
			cin>>x>>y>>temp;
			map[y][x]=temp;
			map[x][y]=temp;//将管道成本放到数组map(即邻接矩阵结构)		
		} 
		 int sum=0;
		 sum=prim();
		 if(sum==-1)
		 	cout<<"?"<<endl;
		else
			cout<<sum<<endl;
	}	
	return 0;
}