最小生成树

//本代码解决，一个图的最小生成树的各边权值总和 

#include<iostream>
using namespace std;
//static声明静态变量，只运行一次， const限定一个变量不允许被改变，产生静态作用。
static const int MAX = 100;
static const int INFTY = (1<<21);
static const int WHITE = 0;
static const int GRAY = 1;
static const int BLACK = 2;

int n, M[MAX][MAX];

int prim() {
    int u, minv;
    /*
    1.d[v]记录连接内顶点与V - T 内顶点的边中，权值最小的边的权值
    2.p[v]记录MST中顶点的父节点 
    */
    int d[MAX], p[MAX], color[MAX];
    
    for(int i = 0; i < n; i++) {
        d[i] = INFTY;
        p[i] = -1;
        color[i] = WHITE;
    }
    
    d[0] = 0;
    
    while(1) {
        minv = INFTY;
        u = -1;
        for(int i = 0; i < n; i++) {
            if(minv > d[i] && color[i] != BLACK){
                u = i;
                minv = d[i];
            }
        }
        //所有点都被遍历结束循环 
        if(u == -1)    break;
        
        color[u] = BLACK;
        //更新与之相关的顶点 
        for(int v = 0; v < n; v++) {
            if(color[v] != BLACK && M[u][v] != INFTY) {
                if(d[v] > M[u][v]) {
                    d[v] = M[u][v];
                    p[v] = u;
                    color[v] = GRAY;
                }
            }
        }
    }
    
    int sum = 0;
    for(int i = 0; i < n; i++) {
        if(p[i] != -1) sum += M[i][p[i]]; //这里为什么不用sum+=d[i]呢，因为害怕它不是连通图 
    }
    return sum;
}

int main() {
    cin >> n;
    //构建邻接矩阵 
    for(int i = 0; i < n; i++) {
        for(int j = 0; j < n; j++) {
            int e;
            cin >> e;
            M[i][j] = (e == -1) ? INFTY: e; 
        }
    }
    
    cout << prim() << endl;
    return 0;
} 
/*
5
-1 2 3 1 -1
2 -1 -1 4 -1
3 -1 -1 1 1
1 4 1 -1 3
-1 -1 1 3 -1
*/