贪心算法之最小生成树Prime C++实现

问题描述】Prim算法解决的是带权重的无向图上连接所有顶点的耗费最小的生成树。Q使用最小堆数据结构。

【输入形式】在屏幕上输入顶点个数和连接顶点间的边的权矩阵。

【输出形式】顺序输出按照贪心选择得到的各顶点序号，及该顶点的前驱顶点序号，及路径长度。

【样例1输入】

8

0 15 7 0 0 0 0 10

15 0 0 0 0 0 0 0

7 0 0 9 12 5 0 0

0 0 9 0 0 0 0 0

0 0 12 0 0 6 0 0

0 0 5 0 6 0 14 8

0 0 0 0 0 14 0 3

10 0 0 0 0 8 3 0

【样例1输出】

3 1 7

6 3 5

5 6 6

8 6 8

7 8 3

4 3 9

2 1 15

【样例说明】

输入：顶点个数为8。连接顶点间边的权矩阵大小为8行8列，位置[i,j]上元素值表示第i个顶点到第j个顶点的距离，0表示两个顶点间没有边连接。

输出：顺序输出按照贪心选择得到的各顶点序号，及该顶点的前驱顶点序号，及路径长度。

【评分标准】根据输入得到准确的输出。

【满分代码】

#include<iostream>
using namespace std;
#define INF 0X7fffffff;


void Prime(int n, int c[][100])
{
    int lowcost[50];
    int closest[50] = {1} ;
    bool s[50] = {false};

    s[0] = true;
    for(int i = 1; i < n; i++)
    {
        lowcost[i] = c[0][i];
    }
    //求邻接点
    for(int i = 0; i < n; i++)
    {
        float min = INF;
        int j = 1;
        for(int k = 1; k < n; k++)
        {
            if((lowcost[k]<min&&(!s[k]))&&lowcost[k] != 0)
            {
                min = lowcost[k];
                j = k;
            }
        }
        if(i < n-1) cout << j+1 << " " << closest[j]+1 << " " << c[j][closest[j]] << endl;
        s[j] = true;

        for(int k = 1; k < n; k++)
        {
            if((!s[k])&&c[j][k]!=0)
            {
                if (c[j][k] < lowcost[k] || lowcost[k] == 0)
                {
                    lowcost[k] = c[j][k];
                    closest[k] = j;
                }
            }
        }

    }

}

int main()
{
    int n;
    int c[100][100];
    cin >> n;
    for(int i = 0; i < n; i++)
        for(int j = 0; j < n; j++)
            cin >> c[i][j];
    Prime(n,c);
}