HDU1102 prim 最小生成树

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#define inf 999999999

int N,M;       // N 代表点的数量，M代表已修建的路的条数
int map[105][105];
int dis[105];
int visit[105];

int prim()
{
for(int i=1;i<=N;i++)  //先把每个点到已经建好的图中的路径置为无穷大
dis[i]=inf;
dis[1]=0;              //起点为顶点1
for(int j=1;j<=N;j++)
{
int t=inf,pos;  //临时变量:最短路径距离t,和该顶点pos
for(int i=1;i<=N;i++)     //查找最短路径'点'
{
if(!visit[i]&&t>dis[i])
{
t=dis[i];
pos=i;
}
}
visit[pos]=1;         //最短路径点标记为已经访问
for(int i=1;i<=N;i++)  //更新最短路径点
{
//如果,顶点i未被访问过
//且图中新加入的pos点到i点的距离短于原来图到i点的距离
//且pos到i的点有连通(即不为无穷大),则将i点到图的距离更新.
if(!visit[i]&&dis[i]>map[pos][i]&&map[pos][i]!=inf)
{
dis[i]=map[pos][i];
}
}
}
int temp=0;
for(int k=1;k<=N;k++)
{
temp+=dis[k];
}
return temp;
}

int main()
{
int a,b;
while(scanf("%d",&N)!=EOF)
{
memset(map,0x3f,sizeof(map));
memset(visit,0,sizeof(visit));
for(int i=1;i<=N;i++)        //构建图的矩阵
{
for(int j=1;j<=N;j++)
{
scanf("%d",&map[i][j]);
}
}
scanf("%d",&M);
for(int i=1;i<=M;i++)       //输入已经修建好的路
{
scanf("%d%d",&a,&b);
map[a][b]=0;
map[b][a]=0;
}
int ans=prim();
printf("%d\n",ans);
}
return 0;
}