最小生成树之Prim算法(最原始最详细入门)

  1 //算法6.8 普里姆算法
  2 #include <iostream>
  3 using namespace std;
  4 
  5 typedef char VerTexType;
  6 typedef int ArcType;
  7 #define MVNum 100
  8 #define MaxInt 32767                        //表示极大值,即∞
  9 
 10 //辅助数组的定义,用来记录从顶点集U到V-U的权值最小的边
 11 struct{
 12     VerTexType adjvex;                        //最小边在U中的那个顶点
 13     ArcType lowcost;                        //最小边上的权值
 14 }closedge[MVNum];
 15 
 16 //- - - - -图的邻接表存储表示- - - - -
 17 typedef char VerTexType;                      //假设顶点的数据类型为字符型
 18 typedef int ArcType;                          //假设边的权值类型为整型
 19 typedef struct{
 20     VerTexType vexs[MVNum];                    //顶点表,一维数组
 21     ArcType arcs[MVNum][MVNum];              //邻接矩阵,表示i-j边上的权值
 22     int vexnum,arcnum;                        //图的当前顶点数和边数
 23 }AMGraph;
 24 
 25 int mincost;         //mincost表示最小生成树所有路径之和的最小值
 26 bool vis[MVNum];    //标记已经归纳到集合U中
 27 int LocateVex(AMGraph G , VerTexType v){
 28     //确定点v在G中的位置,即在顶点数组vexs中查找顶点v的下标
 29     for(int i = 0; i < G.vexnum; ++i)
 30         if(G.vexs[i] == v)
 31             return i;
 32         return -1;//找不到就返回-1
 33 }//LocateVex
 34 
 35 void CreateUDN(AMGraph &G){
 36     //采用邻接矩阵表示法,创建无向网G
 37     int i , j , k;
 38     cout <<"请输入总顶点数,总边数,以空格隔开:";
 39     cin >> G.vexnum >> G.arcnum;                            //输入总顶点数,总边数
 40     cout << endl;
 41 
 42     cout << "输入点的名称,如a或1" << endl;
 43 
 44     for(i = 0; i < G.vexnum; ++i){
 45         cout << "请输入第" << (i+1) << "个点的名称:";
 46         cin >> G.vexs[i];                                    //依次输入点的信息
 47     }
 48     cout << endl;
 49     for(i = 0; i < G.vexnum; ++i)                            //初始化邻接矩阵,边的权值均置为极大值MaxInt
 50         for(j = 0; j < G.vexnum; ++j)
 51             G.arcs[i][j] = MaxInt;
 52     cout << "输入边依附的顶点及权值,如a b 5" << endl;
 53     for(k = 0; k < G.arcnum;++k){                            //构造邻接矩阵
 54         VerTexType v1 , v2;
 55         ArcType w;
 56         cout << "请输入第" << (k + 1) << "条边依附的顶点及权值:";
 57         cin >> v1 >> v2 >> w;                                //输入一条边依附的顶点及权值
 58         i = LocateVex(G, v1);  j = LocateVex(G, v2);        //确定v1和v2在G中的位置,即顶点数组的下标
 59         G.arcs[i][j] = w;                                    //边<v1, v2>的权值置为w
 60         G.arcs[j][i] = G.arcs[i][j];                        //置<v1, v2>的对称边<v2, v1>的权值为w
 61     }//for
 62 }//CreateUDN
 63 
 64 int Min(AMGraph G){
 65     //返回权值最小的点
 66     int index = -1;
 67     int mina = MaxInt;
 68     for(int i = 0 ; i < G.vexnum ; ++i){
 69         if(!vis[i] && mina > closedge[i].lowcost ){//在剩下的集合V-U中选择距离U集合最近的顶点,即权值最小
 70             mina = closedge[i].lowcost;
 71             index = i;
 72         }
 73     }//for
 74     return index;//返回该下标
 75 }//Min
 76 
 77 void MiniSpanTree_Prim(AMGraph G, VerTexType u){//加点法
 78     //无向网G以邻接矩阵形式存储,从顶点u出发构造G的最小生成树T,输出T的各条边
 79     int k , j , i;
 80     mincost = 0;
 81     VerTexType u0 , v0;
 82     k =LocateVex(G, u);                         //k为顶点u的下标
 83     for(j = 0; j < G.vexnum; ++j){              //对V-U的每一个顶点vi,初始化closedge[i]
 84         if(j != k){
 85             vis[j]=false;
 86             closedge[j].adjvex = u;          //默认除u外的各个顶点到U集合中的顶点u的权值最小
 87             closedge[j].lowcost = G.arcs[k][j];  //{adjvex, lowcost}
 88         }//if
 89     }//for
 90     vis[k]=true;closedge[k].lowcost=0;     //初始,U = {u},标记u已经归纳到集合中,自己到自己的权值为0
 91     for(i = 1; i < G.vexnum; ++i){         //选择其余n-1个顶点,生成n-1条边(n= G.vexnum)
 92         k = Min(G);
 93         if(k==-1)break;               //k如果是-1,表示最小生成树已经建立
 94         //求出T的下一个结点:第k个顶点,closedge[k]中存有当前最小边
 95         u0 = closedge[k].adjvex;        //u0为最小边的一个顶点,u0∈U
 96         v0 = G.vexs[k];              //v0为最小边的另一个顶点,v0∈V-U
 97         cout << "" <<u0 << "--->" << v0 << "   权值为" << closedge[k].lowcost << endl;//输出当前的最小边(u0, v0)及其权值
 98         vis[k]=true;mincost+=closedge[k].lowcost;           //第k个顶点并入U集,标记此时顶点k已经归纳到集合U中
 99         for(j = 0; j < G.vexnum; ++j)
100             if(!vis[j] && G.arcs[k][j] < closedge[j].lowcost){//新顶点并入U后在V-U中重新选择最小边
101                 closedge[j].adjvex = G.vexs[k];  //表示顶点j到顶点k还有比原来j到集合U中有最小权值
102                 closedge[j].lowcost = G.arcs[k][j];
103             }//if
104     }//for
105 }//MiniSpanTree_Prim
106 
107 int main(){
108     cout << "************算法6.8 普里姆算法**************" << endl << endl;
109     AMGraph G;
110     CreateUDN(G);
111     cout << endl;
112     cout << "无向图G创建完成!" << endl;
113     cout <<endl;
114 
115     cout << "******利用普里姆算法构造最小生成树结果:******" << endl;
116     MiniSpanTree_Prim(G , G.vexs[0]);//传入起始的顶点
117     cout <<endl;
118     cout<<"最小生成树的所有路径之和最小值为"<<mincost<<endl;
119     return 0;
120 }//main

猜你喜欢

转载自www.cnblogs.com/acgoto/p/9097387.html
今日推荐