prim算法模版

int Prim(int graph[][MAX], int n) {
    /* lowcost[i]记录以i为终点的边的最小权值 */
    int lowcost[MAX];
    
    /* mst[i]记录对应lowcost[i]的起点，当mst[i]=0时表示起点i加入生成树 */
    int mst[MAX];
    int i, j, min, minid, sum = 0;

    /* 默认选择1号节点加入生成树，从2号节点开始初始化 */
    for (i = 2; i <= n; i++) {
        /* 最短距离初始化为其他节点到1号节点的距离 */
        lowcost[i] = graph[1][i];

        /* 标记所有节点的起点皆为默认的1号节点 */
        mst[i] = 1;
    }

    /* 标记1号节点加入生成树 */
    mst[1] = -1;

    /* n个节点至少需要n-1条边构成最小生成树 */
    for (i = 2; i <= n; i++) {
        min = MAXCOST;
        minid = -1;

        /* 找满足条件的最小权值边的节点minid */
        for (j = 2; j <= n; j++) {
            /* 边权值较小且不在生成树中 */
            if (lowcost[j] < min && mst[j] != -1) {
                min = lowcost[j];
                minid = j;
            }
        }
        if(minid != -1){
            /* 输出生成树边的信息:起点，终点，权值 */
            ///printf("%c - %c : %d\n", mst[minid] + 'A' - 1, minid + 'A' - 1, min);
            /* 累加权值 */
            sum += min;

            /* 标记节点minid加入生成树 */
            mst[minid] = -1;

            /* 更新当前节点minid到其他节点的权值 */
            for (j = 2; j <= n; j++) {
                /* 发现更小的权值 */
                if (graph[minid][j] < lowcost[j] && mst[j] != -1) {
                    /* 更新权值信息 */
                    lowcost[j] = graph[minid][j];

                    /* 更新最小权值边的起点 */
                    mst[j] = minid;
                }
            }
        }
    }
    /* 返回最小权值和 */
    return sum;
}