POJ 1287题

//解题思路：朴素的Prim算法
#include <stdio.h>
#include <string.h>    //引入memset的头文件
#define arraysize 51
int maxData = 0x7fffffff;  //定义最大值
int dis[arraysize][arraysize]; //定义邻接矩阵
bool final[arraysize];   //标识该点是否在最小生成树中
int d[arraysize];    //定义最小距离
int P,R;
void Prim()
{
 int i,j;
 int min,v;
 int sum = 0;
 memset(final,0,sizeof(final)); //final初始化
 for(i=1;i<P+1;++i)
 {
  d[i] = dis[1][i];   //最小距离初始化
 }
 final[1] = true;
 d[1] = 0;
 for(i=1;i<P;++i)
 {
  min = maxData;
  for(j=1;j<P+1;++j)
  {
   if(!final[j] && d[j]<min && d[j]<maxData) //此处别忘了判断小于最大距离
   {
    min = d[j];
    v = j;
   }
  }
  final[v] = true;
  sum += min;
  for(j=1;j<P+1;++j)
  {
   if(!final[j] && d[j]>dis[v][j] && dis[v][j]<maxData) //此处别忘了判断小于最大距离
    d[j] = dis[v][j];
  }
 }
 printf("%d\n",sum);
}
int main()
{
 int i,j;
 int start,end,length;
 //freopen("1.txt","r",stdin);
 while(scanf("%d",&P)!=EOF)
 {
  if(P == 0)
   break;
  //临界矩阵初始化
  for(i=1;i<P+1;++i)
  {
   for(j=1;j<P+1;++j)
   {
    dis[i][j] = maxData;
   }
  }
  scanf("%d",&R);
  for(i=1;i<R+1;++i)
  {
   scanf("%d%d%d",&start,&end,&length);
   //判断重边的情况
   if(dis[start][end]>length)
   {
    dis[start][end] = length; //无向图
    dis[end][start] = length;
   }
  }
  Prim();
 }
 return 0;
}