POJ1258--贪心&最小生成树的prim算法

用prim算法实现，T1集合为所有点集合,T2集合表示已经是最小生成树中的点
lowcost[maxn]用来实现集合T1内各顶点到集合T2中各顶点的权值最小边的权值
nearvex[maxn]用来实现集合T1内顶点距集合T2中哪个顶点的距离最小

prim算法的思想：

初始：lowcost[k]=edge[v0][k] , nearvex[k]=v0;其中v0指的是从哪点来构造最小生成树。

当lowcost[i]=-1时我们让它表示i顶点加入到了T2集合，

什么样的情况下我们将i结点加入到T2集合呢？

我们选择那些在T1集合中，距离T2集合中点最小的lowcost[k],将该点k加入到集合T2中，

加入后我们应该改变那些在T1集合中点到T2集合中的最小权值，如果edge[k][i]<lowcost[i]

则将lowcost[i]=edge[k][i];nearvex[i]=k;这就完成了一次操作，进行n-1次后，就能构成一个最小生成树

#include <stdio.h>
#define inf 100000
#define maxn 401
int n;
int lowcost[maxn];
int nearvex[maxn];
int edge[maxn][maxn];
int prim(int u0)
{
 int sumweight = 0;
 int i, j;
 for(i = 1; i <= n; i++)  //顶点从1开始
 {
  lowcost[i] = edge[u0][i];
  nearvex[i] = u0;
 }
 lowcost[u0] = 0;
 nearvex[u0] = -1;   //lowcost[]为-1时表示该点已经加入T2
 for(i = 1; i < n; i++)  //将n-1个顶点加入到顶点集合T2
 {
  int min = inf;
  int v = -1;
  for(j = 1; j <= n; j++)
  {
   if(nearvex[j] != -1 && lowcost[j] < min)
   {
    min = lowcost[j];
    v = j;
   }
  }
  if(v != -1)
  {
   //printf("%d %d %d\n", nearvex[v], v, lowcost[v]);
   nearvex[v] = -1;   //把v加入T2中
   sumweight += lowcost[v];
   for(j = 1; j <= n; j++)
   {
    //更新lowcost[]数组
    if(nearvex[j] != -1 && edge[v][j] < lowcost[j])//j在T1集合中&&并且满足v到j的距离小于j到T2中顶点的最小值
    {
     lowcost[j] = edge[v][j];
     nearvex[j] = v;
    }
   }
  }
 }
 return sumweight;
}
int main()
{
 int i, j;
 int result;
 while(scanf("%d", &n) != EOF)
 {
  for(i = 1; i <= n; i++)
   for(j = 1; j <= n; j++)
   {
    scanf("%d", &edge[i][j]);
   }
  result = prim(1);
  printf("%d\n", result);
 }
 return 0;
}