【PAT数据结构与算法题目集】公路村村通（最小生成树）

【PAT数据结构与算法题目集】公路村村通（最小生成树）

题目

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

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

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

输入样例:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:

12

思路

这是个最小生成树问题，此处利用Kruskal算法。

代码

#include <iostream>
#include <stdio.h>
#include <string>
#include<string.h>
#include <vector>
#include <map>
#include <stack>
#include <algorithm>
#include <stack>
#include <queue>
using namespace std;

int matrix[1005][1005];
int parent[1005];

typedef struct edge{
    int l;
    int r;
    int weight;
};

edge edges[4000];

bool compare(edge a, edge b) {
    return a.weight < b.weight;
}

int find_parent(int i) {
    while(parent[i] != i) {
        i = parent[i];
    }
    return i;
}

int main() {
    int i, j, k, n, m, a, b, ans, edge_number;
    while(~scanf("%d%d", &n, &m)) {
        memset(matrix, 0, sizeof(matrix));
        memset(edges, 0, sizeof(edges));
        memset(parent, 0, sizeof(parent));

        for(i=1; i<=n; i++) {
            parent[i] = i;
        }

        for(a=0; a<m; a++) {
            scanf("%d %d %d", &i, &j, &k);
            if(i > j) {
                b = i;
                i = j;
                j = b;
            }
            matrix[i][j] = matrix[j][i] = k;
            edges[a].l = i;
            edges[a].r = j;
            edges[a].weight = k;
        }
        sort(edges, edges + m, compare);

        ans = 0;
        edge_number = 0;

        for(i=0; i<m; i++) {
            if(edge_number == n-1) {
                break;
            }
            a = find_parent(edges[i].l);
            b = find_parent(edges[i].r);
            if(a == b) {
                continue;
            }
            ans = ans + edges[i].weight;
            edge_number++;
            parent[a] = b;
        }

        if(edge_number < n-1) {
            ans = -1;
        }
        cout<<ans<<endl;

    }
    return 0;
}