数据结构——Prim算法

看了一下表，刚刚过了12点，今天就是年三十了，刚看完辽视春晚，大晚上睡不着，就把白天复习的Prim给更一下。我只计算了最小生成树的权值和，没有记录出生成图各边情况，其实我代码中，Edge数组存放的是某未被访问节点（i）到已被访问的所有节点中最近的距离，可以改一下，定义一个结构体，里面放有起点，终点，和Edge[i]值即可。由访问的顺序依次压入队列，输出时挨个取top再pop就可以了。

老规矩，在网上随便盗了个图测试代码、、、

C++代码：

#include<iostream>
#include<limits.h>
#include<queue>

#define MAXVEX 100

using namespace std;

typedef struct
{
    int matrix[MAXVEX][MAXVEX];
    int numNodes, numEdges;
} Graph;

void CreateGraph(Graph *Gp)
{
    int i, j, k, w;
    bool isDirected;
    cout<<"无向图请输入0，有向图请输入1"<<endl;
    cin>>isDirected;
    cout<<"请输入顶点数和边数（空格分隔）："<<endl;
    cin>>Gp->numNodes>>Gp->numEdges;;
    for (i=1;i<=Gp->numNodes;i++)
    {
        for (j=1;j<=Gp->numNodes;j++)
        {
            if (i==j)
                Gp->matrix[i][j]=0;
            else
                Gp->matrix[i][j]=INT_MAX;
        }
    }
    cout<<"请输入边（vi, vj)和权值w,三个变量空格分隔即可："<<endl;
    for (k=0;k<Gp->numEdges;k++)
    {
        cin>>i>>j>>w;
        Gp->matrix[i][j]=w;
        if(!isDirected)
            Gp->matrix[j][i]=Gp->matrix[i][j];
    }
}

void ShowGraph(Graph M){
    int i, j;
    for(i=1;i<=M.numNodes;i++){
        for(j=1;j<=M.numNodes;j++){
            if(M.matrix[i][j]!=INT_MAX)
                cout<<M.matrix[i][j]<<" ";
        else
            cout<<"*"<<" ";
        }
        cout<<endl;
    }
}

void Prim(Graph M, int start){
    int Visited_count=0, sum=0; //分别为已被访问的节点数和总权值
    bool *isVisited=new bool[M.numNodes+1];//判断各点是否被访问过
    int *Edge=new int[M.numNodes+1]; //记录各点（如果未被访问的话）到已被访问的各点中某一点的最小距离
    for(int i=1;i<=M.numNodes;i++)
        isVisited[i]=false;
    isVisited[start]=true;
    for(int i=1;i<=M.numNodes;i++){
        Edge[i]=M.matrix[start][i];
    }
    int last_Node;//最近由未访问状态转至以访问状态的点的序号
    Visited_count++;
    while(Visited_count<M.numNodes){
        int min_Edge=INT_MAX;
        for(int i=1;i<=M.numNodes;i++){
            if(!isVisited[i] && Edge[i]<min_Edge){
                min_Edge=Edge[i];
                last_Node=i;
            }
        }
        Visited_count++;
        isVisited[last_Node]=true;
        sum+=min_Edge;
        for(int i=1;i<=M.numNodes;i++){ //因为由刚刚发现的点，更新Edge数组中的值
            if(!isVisited[i] && M.matrix[last_Node][i]<Edge[i]){
                Edge[i]=M.matrix[last_Node][i];
            }
        }
    }
    cout<<"最小生成树总长度为"<<sum<<endl;
    delete[]isVisited;
    delete[]Edge;
}

int main(){
    Graph M;
    CreateGraph(&M);
    cout<<"邻接矩阵为："<<endl;
    ShowGraph(M);

    int start;
    cout<<"请输入起始点"<<endl;
    cin>>start;
    Prim(M, start);

    return 0;
}

运行结果：（序号依次往后窜了1，V0用1表示，以此类推）