hdoj1102 Constructing Roads(最小生成树)

来源:http://acm.hdu.edu.cn/showproblem.php?pid=1102

求最小生成树，我这里用的是prim算法(数据结构里的背过的模版)

将已经有路的村庄之间的距离设为0.

代码如下:

#include<stdio.h>
#include<iostream>
#include<string>
#define INF 0x7ffffff
using namespace std;
int main()
{
	int map[101][101];
	int lowcost[101],vis[101];
	int n,a,b,num,i,j,min,t,all;
	while(~scanf("%d",&n))
	{
		for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
		scanf("%d",&map[i][j]);
		scanf("%d",&num);
		for(i=0;i<num;i++)
		{
			scanf("%d%d",&a,&b);
			map[a][b]=map[b][a]=0;
		}
		memset(vis,0,sizeof(vis));
		for(i=1;i<=n;i++) 
		lowcost[i]=map[1][i];
		vis[1]=1;all=0;
		for(i=2;i<=n;i++)
		{
			min=INF;
			for(j=1;j<=n;j++)
			{
				if(min>lowcost[j]&&!vis[j])
				{
					min=lowcost[j];
					t=j;
				}
			}
			vis[t]=1;
			all+=min;
			for(j=1;j<=n;j++)
			{
				if(lowcost[j]>map[t][j]&&!vis[j])
				lowcost[j]=map[t][j];
			}
		}
		printf("%d\n",all);
	}
	return 0;
}