Prim算法

cost数组表示的是图的加权值。其中顶点到自身的权值为0，到与自身相接的顶点为它们之间的权值，不直接相连的为INT_MAX。而tree数组准确描述应该是最小生成树的边集，因为一个N个顶点的图中，其最小生成树的边集中的元素只能为N-1。否则如果如果边集大于N-1，那么一定存在回路，如果小于N-1那么一定有一个点不在该树种。near集合中存储的是还未进入最小生成树中的点集。

Prim算法的过程如下：

1）从顶点集合中根据加权值集合找到权值最小的一条边。

2）初始化near集合，判定其他点i到1）中所得的那条边的两个顶点之间距离，将两点中最近的点作为near[i]的初始值，表示未进入最小生成树中的点距离最小生成树最近的点。同时将上述的那条边的两个顶点从near集合中删除（代码中用-1表示）。

3）选择出与最小生成树最近且未进入最小生成树的点，并将其从near集合中删除。

4）修改那些未进入最小生成树中的点到最小生成树中最近的点，即修改near集合。

#include <iostream>
#include <limits>
using namespace std;

const int N = 6;
int cost[N][N] = {{0      ,10     ,INT_MAX,30     ,45     ,INT_MAX},
                  {10     ,0      ,50     ,INT_MAX,40     ,25     },
                  {INT_MAX,50     ,0      ,INT_MAX,35     ,15     },
                  {30     ,INT_MAX,INT_MAX,0      ,INT_MAX,20     },
                  {45     ,40     ,35     ,INT_MAX,0      ,55     },
                  {INT_MAX,25     ,15     ,20     ,55     ,0      }};
int near[N];

int prim(int** tree){
    int k=0, l=1;
    int min = cost[k][l];
    for(int i = 0 ; i < N ; ++i){
      for(int j = i+1; j < N; ++j){
        if(cost[i][j]<min){
          min = cost[i][j];
          k = i;
          l = j;
        }
      }
    }
    tree[0][0] = k, tree[0][1] = l;
    
    for(int i = 0; i < N ; ++i)
      near[i] = (cost[i][k]<cost[i][l])?k:l;
    near[k] = near[l] = -1; // add the vertex k and l to the MST
    
    for(int i = 1; i < N-1; ++i){
      int tmpMin = INT_MAX;
      int flag = 0;
      for(int j = 0; j < N; ++j){
        if(near[j]!=-1 && tmpMin>cost[j][near[j]]){
          tmpMin = cost[j][near[j]];
          flag = j;
        }
      }
      tree[i][0] = flag, tree[i][1] = near[flag];
      near[flag] = -1;
      for(int h = 0; h < N; ++h){
        if(near[h]!=-1 && cost[h][near[h]]>cost[flag][h])
          near[h] = flag;
      }
    }
}

int main(){
    int **tree = new int*[N-1];
    for(int i = 0; i < N-1; i++)
      tree[i] = new int[2];
    prim(tree);
    for(int i = 0 ; i < N-1 ; i++)
      cout<<tree[i][0]<<"->"<<tree[i][1]<<endl;
    delete[] tree;
    system("PAUSE");
}