最小生成树 prim()

/**
    prim算法应该是我学会的第一个图算法

    最小生成树，一个无向图使每两个点都连通的最小花费，每条边都会有
    边权，这个算法在点少边多的情况下要比kruskal 更有优势，一般都不多于
    1000个点。

    比如：几个村庄铺设电话线，给出了一个图，图上描述了每两个村庄进行连接的花费，
    现在求一种最小花费使每两个村庄有且仅有一条路径相连。

    还是那句话，图算法现成模板太多，关键在于把问题转换成这个模型

    在里面标记，可得到最小生成树的形状，不只是最小权

    有向图的最小生成树叫最小树形图，我空间里面有，在前面，比这个复杂点
*/

int map[Max][Max]; // map = inf; 初始化为正无穷大

int n; //点的个数

int prim() {
    int dis[Max], count = n-1, i, min_edge, min_node;
    for (i=1; i<=n; i++)
        dis[i] = inf;
    int now = s; //设s 为起点随便找个 (1 - n)
    int ans = 0;
    while (count--) {
        dis[now] = -1;
        min_edge = inf;
        for (i=1; i<=n; i++) {
            if (now!=i && dis[i] >= 0) {
                dis[i] = min(dis[i], map[now][i]);
                if (dis[i] < min_edge) {
                    min_edge = dis[i];
                    min_node = i;
                }
            }
        }
        now = min_node;
        ans += min_edge;
    }
    return ans;
}

int main() {

    //建好图 后直接调用

    prim();

    return 0;
}

收藏于 2012-01-08
来自于百度空间