poj Agri-Net 最小生成树 prim 算法

最小生成树, prim 算法

普里姆算法（Prim算法），图论中的一种算法，可在加权连通图里搜索最小生成树.

下面对算法的图例描述(转载 的图  , 用于理解) ; 其原理 类似于 Dijstra算法最短路算法  

 

图例	说明	不可选	可选	已选（Vnew）
	此为原始的加权连通图。每条边一侧的数字代表其权值。	-	-	-
	顶点D被任意选为起始点。顶点A、B、E和F通过单条边与D相连。A是距离D最近的顶点，因此将A及对应边AD以高亮表示。	C, G	A, B, E, F	D
	下一个顶点为距离D或A最近的顶点。B距D为9，距A为7，E为15，F为6。因此，F距D或A最近，因此将顶点F与相应边DF以高亮表示。	C, G	B, E, F	A, D
	算法继续重复上面的步骤。距离A为7的顶点B被高亮表示。	C	B, E, G	A, D, F
	在当前情况下，可以在C、E与G间进行选择。C距B为8，E距B为7，G距F为11。E最近，因此将顶点E与相应边BE高亮表示。	无	C, E, G	A, D, F, B
	这里，可供选择的顶点只有C和G。C距E为5，G距E为9，故选取C，并与边EC一同高亮表示。	无	C, G	A, D, F, B, E
	顶点G是唯一剩下的顶点，它距F为11，距E为9，E最近，故高亮表示G及相应边EG。	无	G	A, D, F, B, E, C
	现在，所有顶点均已被选取，图中绿色部分即为连通图的最小生成树。在此例中，最小生成树的权值之和为39。	无	无	A, D, F, B, E, C, G

 

题目地址

点击打开链接

代码实现 

#include<stdio.h>
#include<iostream>

using namespace std;
const int inf=1<<29;
const int N= 1000;
int map[N][N];//邻接矩阵 
int dist[N];// 到每个点的距离 
int pre[N];//结点 
int n;
int sum;
void pirm(int v0)
{
	int i,j;
	for(i=1;i<=n;i++)
	{
		dist[i]=map[v0][i];
		pre[i]=i;		
	}
	pre[v0]=0;
	for(i=0;i<n-1;i++)
	{
		int mid=inf;
		int u=-1;
		for(j=0;j<=n;j++)
		{
			if(dist[j]&&dist[j]<mid)
			{
				mid=dist[j];
				u=j;
			}
		}
		sum+=dist[u];//计算路径之和 
		dist[u]=0;
		for(j=0;j<=n;j++)
		{
			if(map[u][j]<dist[j])
			{
				dist[j]=map[u][j];	
				pre[j]=u;
			}	
		}	
	}
	
}
int main()
{
	while(~scanf("%d",&n))
	{
		for(int i=0;i<=n;i++)//初始化 
			for(int j=0;j<=n;j++)
				map[i][j]=inf;
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				scanf("%d",map[i]+j);//直接输入   or 
		sum=0;
		pirm(1);//以1为起点 
		printf("%d\n",sum);
	}
	return 0;
}