最小生成树 Prim算法

Description

 有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

 

Input

 输入包含多组数据，格式如下。

第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)

剩下m行每行3个非负整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c（城市编号从1到n）。

 

Output

 每组输出占一行，仅输出最小花费。

Sample

Input

3 2
1 2 1
1 3 1
1 0

Output

2
0

代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int INF = 0x3f3f3f3f;
int v[110];   //记录是否已经过
int map1[110][110];
int dis[110];

int prim(int n){
    //初始化dis[]
    for(int i=1; i<=n; i++){
        dis[i] = map1[1][i];
    }
    v[1] = 1;
    dis[1] = 0;

    int sum = 0;

    for(int j=0; j<n-1; j++){
        int min1=INF, flag=0;
        // 寻找未标记的最小值
        for(int i=1; i<=n; i++){
            if(!v[i]&&dis[i]<min1){
                min1 = dis[i];
                flag = i;
            }
        }

        v[flag] = 1;
        sum+=min1;

        // 更新距离
        for(int i=1; i<=n ;i++){
            if(!v[i]&&map1[flag][i]<dis[i]){
                dis[i] = map1[flag][i];
            }
        }
    }

    return sum;
}

int main()
{
    int n, m;
    while(cin>>n>>m){
        memset(v, 0, sizeof(v));  // 初始化v[]
        if(!m) cout<<"0"<<endl;
        else{
            memset(map1, INF, sizeof(map1));
            while(m--){
                int a, b, c;
                cin>>a>>b>>c;
                if(map1[a][b]>c){
                    map1[a][b] = map1[b][a] = c;
                }
            }

            cout<<prim(n)<<endl;

        }
    }

    return 0;
}


//1).输入：一个加权连通图，其中顶点集合为V，边集合为E；
//2).初始化：Vnew = {x}，其中x为集合V中的任一节点（起始点），Enew = {},为空；
//3).重复下列操作，直到Vnew = V：
//a.在集合E中选取权值最小的边<u, v>，其中u为集合Vnew中的元素，而v不在Vnew集合当中，
//并且v∈V（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
//b.将v加入集合Vnew中，将<u, v>边加入集合Enew中；
//4).输出：使用集合Vnew和Enew来描述所得到的最小生成树。