PTA 7-13 军事单位光缆铺设-Prim算法

在多个军事单位之间铺设通信光缆，实现任意两个单位间都能进行通信互联。现已得到拟铺设光缆的各条路线费用，请编写程序计算需铺设哪几条线路，使造价最低且能满足互联互通。

输入格式:

输入的第一行给出军事单位数目N (1≤N≤20)和拟铺设的通信光缆数M

接下来的M行对应各个军事单位间铺设通信光缆的成本，每行给出3个正整数，分别是两个军事单位的编号（从1编号到N），此两单位间铺设光缆的成本。

输出格式:

输出需铺设的光缆，按prim算法从编号1开始得到的顺序，输出每条光缆信息，每行输出一条光缆信息，形式如：单位1编号,单位2编号,费用。(每条光缆信息中编号小的放前面，编号大的放后面，逗号为英文状态下的逗号)。

输入样例:

4 6
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5

输出样例:

1,2,1
1,4,1
2,3,3

代码实现：

#include <stdio.h>
#include <stdbool.h>
#include <limits.h>
#include <stdlib.h>
#define M 20

typedef struct edge_node
{
    int incr_vert, vertex;
    int cost;
} edge;

void Prim2(int c[M][M], int n)
{
    int v, i, j, k;
    edge t, wait[M-1];

    for (v = 0; v < n - 1; v++)
    {
        wait[v].incr_vert = 0;
        wait[v].vertex = v + 1;
        wait[v].cost = c[0][v + 1];
    }

    for (i = 0; i < n - 2; i++)
    {
        k = i;
        for (j = i + 1; j < n - 1; j++)
            if (wait[j].cost < wait[k].cost) k = j;
        t = wait[k]; wait[k] = wait[i]; wait[i] = t;
        v = wait[i].vertex;
        for (j = i + 1; j < n - 1; j++)
            if (c[v][wait[j].vertex] < wait[j].cost)
            {
                wait[j].cost = c[v][wait[j].vertex];
                wait[j].incr_vert = v;
            }
    }

    for (i = 0; i < n - 1; i++)
    {
        if (wait[i].incr_vert > wait[i].vertex) j = wait[i].incr_vert, wait[i].incr_vert = wait[i].vertex, wait[i].vertex = j;
        printf("%d,%d,%d", wait[i].incr_vert + 1, wait[i].vertex + 1, wait[i].cost);
        printf("\n");
    }
    
}

int main()
{
    int n, x2, c[M][M];
    int i, q, w, e;

    for (i = 0; i < M; i++)
        for (q = 0; q < M; q++) c[i][q] = 99;

    scanf("%d %d", &n, &x2);
    for (i = 0; i < x2; i++)
    {
        scanf("%d %d %d", &q, &w, &e);
        c[q - 1][w - 1] = c[w - 1][q - 1] = e;
    }
    Prim2(c, n);
}