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;
}