poj1258_prime

<pre name="code" class="cpp">/*
题意：给你一个n*n的连通图，求将他全部连起来的最小花费，即求其最小生成树

这题虽然用kruskal更合适（下面写了原因），但这里我们用prim算法<pre>
1).输入：一个加权连通图，其中顶点集合为V，边集合为E；
2).初始化：Vnew = {x}，其中x为集合V中的任一节点（起始点），Enew = {},为空；
3).重复下列操作，直到Vnew = V：
a.在集合E中选取权值最小的边<u, v>，其中u为集合Vnew中的元素，而v不在Vnew集合当中，
并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
b.将v加入集合Vnew中，将<u, v>边加入集合Enew中；
4).输出：使用集合Vnew和Enew来描述所得到的最小生成树。
 
prim算法适合稠密图，其时间复杂度为O(n^2)，其时间复杂度与边得数目无关，
而kruskal算法的时间复杂度为O(eloge)跟边的数目有关，适合稀疏图。
*/
#include<stdio.h>
#include<string.h>
#include<limits.h>
#define N 102
//map[][]是给定的带权无向图，low[]记录了每两个点之间当前的最小值，vis[]标记是否被访问过 
int n,map[N][N],low[N],vis[N],ans;
int prime()
{
	memset(vis,0,sizeof vis);//标记都未访问 即 Enew = {} 
	int pos ,min,res=0;//pos记录起点，min保存i到j的最小值，res保存当前最短路 
	//这里假设起点是1
	vis[1]=1;//Vnew = {1} 
	pos=1; 
	//初次给low数组赋值
	for(int i=1;i<=n;i++)
	    if(i!=pos)
		    low[i]=map[pos][i];
    //再运行n-1次 
     for(int i=1;i<n;i++)
	 {
	 	//找出最小权值并记录位置
	 	 min=INT_MAX;
		 for(int j=1;j<=n;j++)
 		      if(!vis[j]&&min>low[j])
			   {
   			       min=low[j];
				   pos=j;	
		       }
       //最小值相加获得最小生成树的
       res+=min;
       //标记该点 
       vis[pos]=1;//Vnew = {1,,,pos}  
       //更新low[]
	   for(int j=1;j<=n;j++)
	       if(!vis[j]&&low[j]>map[pos][j])
		       low[j]=map[pos][j]; 
 	 } 
   return res;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
	{
		//map[][]初始化为 INT_MAX,即认为任何两点都不连接 
		memset(map,INT_MAX,sizeof map);
        for(int i=1;i<=n;i++)
           for(int j=1;j<=n;j++)
                scanf("%d",&map[i][j]);
        ans=prime();
        printf("%d\n",ans);
	}	
}