PTA 最小生成树（普里姆算法)

最新推荐文章于 2024-10-08 20:32:22 发布

vi_tamin_

最新推荐文章于 2024-10-08 20:32:22 发布

阅读量159

点赞数 4

文章标签：算法 c++ 开发语言

本文链接：https://blog.csdn.net/vi_tamin_/article/details/134768764

版权

作者王东单位贵州师范学院

试实现普里姆最小生成树算法。

函数接口定义：

void Prim(AMGraph G, char u);

其中 G 是基于邻接矩阵存储表示的无向图，u表示起点

裁判测试程序样例：

#include <iostream>
#define MVNum 10
#define MaxInt 32767 
using namespace std;

struct edge{
    char adjvex;
    int lowcost;
}closedge[MVNum];

typedef struct{ 
    char vexs[MVNum];   
    int arcs[MVNum][MVNum]; 
    int vexnum,arcnum;
}AMGraph;
int LocateVex(AMGraph G , char v);//实现细节隐藏
int Min(AMGraph G);//实现细节隐藏
int CreateUDN(AMGraph &G);//实现细节隐藏

void Prim(AMGraph G, char u);

int main(){
    AMGraph G;
    CreateUDN(G);
    char u;
    cin >> u;
    Prim(G , u);
    return 0;
}

/* 请在这里填写答案 */

输入样例：

第1行输入结点数vexnum和边数arcnum。第2行输入vexnum个字符表示结点的值，接下来依次输入arcnum行，每行输入3个值，前两个字符表示结点，后一个数表示两个结点之间边的权值。最后一行输入一个字符表示最小生成树的起始结点。

输出样例：

按最小生成树的生成顺序输出每条边。

0->5
5->4
4->3
3->2
2->1
1->6

我的代码：

void Prim(AMGraph G, char u) {
    char c;
    //初始化辅助数组
    int p = 0;
    for (; p < G.vexnum; p++) {
        if (G.vexs[p] == u)break;
    }
    for (int i = 0; i < G.vexnum; i++) {
        closedge[i].adjvex = u;
        if (G.arcs[p][i]&&G.vexs[i]!=u) {
            closedge[i].lowcost = G.arcs[p][i];
        }
        else if (G.vexs[i] == u) {
            closedge[i].lowcost = 0;
        }
        else {
            closedge[i].lowcost = MaxInt;
        }
    }
    int min = MaxInt;
    do {
        min = MaxInt;
        //找到最小值
        for (int i = 0; i < G.vexnum; i++) {
            if (closedge[i].lowcost != 0 && closedge[i].lowcost < min) {
                min = closedge[i].lowcost;
            }
        }
        //输出最小值
        for (int i = 0; i < G.vexnum; i++) {
            if (closedge[i].lowcost == min) {
                closedge[i].lowcost = 0;
                printf("%c->%c\n", closedge[i].adjvex, G.vexs[i]);
                c = G.vexs[i];//记录该点的名称
                p = i;//记录该点的下标
                break;
            }

        }
        //更新辅助数组
        for (int i = 0; i < G.vexnum; i++) {
            if (closedge[i].lowcost == 0)continue;
            else if (closedge[i].lowcost > G.arcs[p][i]) {
                closedge[i].adjvex = c;
                closedge[i].lowcost = G.arcs[p][i];
            }
        }
    } while (min!=MaxInt);//只要最小值更新就不结束循环
}