【最小生成树+Prim】杭电 hdu 1102 Constructing Roads

/* THE PROGRAM IS MADE BY PYY */
/*----------------------------------------------------------------------------//
Copyright (c) 2012 panyanyany All rights reserved.

URL   : http://acm.hdu.edu.cn/showproblem.php?pid=1102 Name  : 1102 Constructing Roads

Date  : Tuesday, February 7, 2012
Time Stage : half an hour

Result:
5325772	2012-02-07 15:30:46	Accepted	1102
15MS	212K	2023 B
C++	pyy

5325749	2012-02-07 15:27:05	Time Limit Exceeded	1102
1000MS	184K	2016 B
C++	pyy

Test Data :

Review :

//----------------------------------------------------------------------------*/

#include <cstdio>
#include <stdlib.h>
#include <string.h>

#include <algorithm>
using namespace std ;

#define MEM(a, v)		memset (a, v, sizeof (a))	// a for address, v for value
#define max(x, y)		((x) > (y) ? (x) : (y))
#define min(x, y)		((x) < (y) ? (x) : (y))

#define INF		(0x3f3f3f3f)
#define MAXN	(103)
#define MAXE	(MAXN*(MAXN-1)/2)

#define DEBUG	/##/

int		n, q ;
int		map[MAXN][MAXN], dist[MAXN], used[MAXN] ;

int Prim()
{
int i, j, sum, iMinPath, MinPath ;
// 一开始把1放入集合A中，然后求集合A到集合B各点的距离
for (i = 1 ; i <= n ; ++i)
dist[i] = map[1][i] ;

MEM(used, 0) ;
sum = 0 ;
used[1] = true ;	// 标记集合A中的点
for (i = 1 ; i <= n-1 ; ++i)
{
MinPath = INF ;
// 求出集合A到集合B中各点距离的最小值
for (j = 1 ; j <= n ; ++j)
if (!used[j] && dist[j] < MinPath)
{
MinPath = dist[j] ;
iMinPath = j ;
}

// 将集合B中的点并入集合A中
used[iMinPath] = true ;
// 新点加入意味着新的路径产生了
sum += MinPath ;

// 由于新点的加入，集合A到集合B中各点的距离可能会变化，
// 所以要及时更新
for (j = 1 ; j <= n ; ++j)
if (!used[j] && dist[j] > map[iMinPath][j])
dist[j] = map[iMinPath][j] ;
}
return sum ;
}

int main()
{
int i, j ;
int x, y ;
while (scanf("%d", &n) != EOF)	// 一开始忘了判断 EOF 了，= =！
{
MEM (map, INF) ;
for (i = 1 ; i <= n ; ++i)
for (j = 1 ; j <= n ; ++j)
scanf ("%d", &map[i][j]) ;
scanf ("%d", &q) ;
for (i = 1 ; i <= q ; ++i)
{
scanf ("%d%d", &x, &y) ;
map[x][y] = map[y][x] = 0 ;	// 已建成的路就不用再建了
}
printf ("%d\n", Prim()) ;
}
return 0 ;
}