公路村村通

7-10 公路村村通 (30 分)

现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:

输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

#include <iostream>
using namespace std;
//树的多个分支直线同一树根要注意；
int N, D;
int cost[1005][1005];
int parent[1005];
int disk[1005];
int money1 = 0;
void init(){
    for(int i = 1; i <= N; i++){
        parent[i] = -1;
        disk[i] = 100000;
    }
    for(int i = 1; i <= N; i++)
        for(int j = 1; j <= N; j++)
            cost[i][j] = 100000;
}
int prim(){
    disk[1] = 0;
    for(int i = 1; i <= N; i++){
        if(cost[1][i] < disk[i]){
            disk[i] = cost[1][i];
            parent[i] = 1;
        }
        
    }
    int v;
    while(1){
        int Disk = 100000;
        int flag = 0;
        for(int i = 1; i <= N; i++){
            if(disk[i] != 0 && disk[i] < Disk){
                Disk = disk[i];
                v = i;
                flag = 1;
            }
        }
        if(flag == 0) break;
        disk[v] = 0;
        money1 += Disk;
        for(int i = 1; i <= N; i++){
            if(disk[i] != 0 && cost[v][i] < disk[i]){
                disk[i] = cost[v][i];
                parent[i] = v;
            }
        }
    }
    for(int i = 1; i <= N; i++)
        if(disk[i] != 0)
            return 0;
    return v;
}
int main()
{
    cin >> N >> D;
    init();
    int L1, L2;
    for(int i = 1; i <= D; i++){
        cin >> L1 >> L2;
        cin >> cost[L1][L2];
        cost[L2][L1] = cost[L1][L2];
    }
    int L = prim();
    if(L != 0){
        cout << money1;
    }
    else
        cout << -1;
}