hdu1102 Constructing Roads--最小生成树prim

原题链接： http://acm.hdu.edu.cn/showproblem.php?pid=1102

题意：

给n个城市，分别是1----n，现在给出两两城市之间距离，求其最小生成树，另外还有m组操作，u v表示城市u与v之间的直接连通的路已经建好，具体操作看代码，很简单的题。

代码：

#define _CRT_SECURE_NO_DEPRECATE 

#include<iostream>
#include<algorithm>
#include<cmath>
#include<string.h>

using namespace std;

int p[105][105];
int lowCost[105];//这里的作用既是作为边权值也是作为标记，相当于visited数组的作用，标记该点是否已加入生成树
int N, M;

int prim()
{
	for (int i = 1; i <= N; i++)
		lowCost[i] = p[1][i];

	lowCost[1] = -1;//1作为起点，赋值为-1标记该点加入生成树
	int ans = 0;
	for (int i = 1; i <= N - 1; i++)
	{
		int min = 99999999;
		int minIndex;
		for (int j = 1; j <= N; j++)
		{
			if (lowCost[j] != -1 && lowCost[j] < min)//该点没有加入生成树且小于min
			{
				min = lowCost[j];
				minIndex = j;
			}
		}

		ans += min;
		lowCost[minIndex] = -1;//标记该点已加入生成树，下面的操作不对该点做处理

		for (int j = 1; j <= N; j++)
		{
			if (p[minIndex][j] < lowCost[j])
				lowCost[j] = p[minIndex][j];
		}
	}

	return ans;
}

int main()
{
	while (~scanf("%d", &N) && N)
	{
		for (int i = 1; i <= N; i++)
			for (int j = 1; j <= N; j++)
				scanf("%d", &p[i][j]);

		int u, v;
		scanf("%d", &M);
		for (int i = 1; i <= M; i++)
		{
			scanf("%d%d", &u, &v);
			p[u][v] = 0;
			p[v][u] = 0;//注意这里！！！
		}

		printf("%d\n", prim());
	}

	return 0;
}