hdu 1102 Constructing Roads(Prime算法)

本题链接：点击打开链接

        本题采用的是另一种算法（Prime算法）算法是采用一个二维数组map其下标分别表示该条路所连接的两个村庄的标号，存放的内容是该路的长度，即权值。使用一个mark数组标记已连过的村庄，一个一维数组lowcost存放是以下标为终点的权值，有些路已连通不再需要修建，便将对应map数组的值标为0（一条路对应两个map）。具体请参见代码：

#include<stdio.h>
#include<string.h>
#define INF 0xffffff
int map[110][110];
int mark[110],lowcost[110];
void prime(int n)
{
	int sum=0;
	memset(mark,0,sizeof(mark));
	for(int i=1;i<=n;i++)
		lowcost[i]=map[1][i];
	mark[1]=1;
	lowcost[1]=0;
	for(int i=1;i<n;i++)//有n个村庄，故需查找n-1次 
	{
		int vir,min=INF;
		for(int j=1;j<=n;j++)//遍历所有点查找到权值最小的点，记录其下标 
		{
			if(!mark[j]&&lowcost[j]<min)
			{
				min=lowcost[j];
				vir=j;
			}
		}
		mark[vir]=1;//将以查找过的标记 
		sum+=lowcost[vir];//求此最小生成树的权值 
		for(int k=1;k<=n;k++)//更新未取点到已取点的权值 
			if(!mark[k]&&lowcost[k]>map[vir][k])
				lowcost[k]=map[vir][k];
	}
	printf("%d\n",sum);
}
int main()
{
	int n,m;
	while(scanf("%d",&n)!=EOF)
	{
		memset(map,INF,sizeof(map));//将图初始化为无穷大 
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				scanf("%d",&map[i][j]);//输入图 
		scanf("%d",&m);
		int x,y;
		while(m--)
		{
			scanf("%d%d",&x,&y);
			map[x][y]=map[y][x]=0;//将已修的路权值标为0 
		}
		prime(n);
	}
	return 0;
}