自来水管道

题目描述

你领到了一个铺设校园内自来水管道的任务。校园内有若干需要供水的点，每两个供水点可能存在多种铺设路径。对于每一种铺设路径，其成本是预知的。 任务要求最终铺设的管道保证任意两点可以直接或间接的联通，同时总成本最低。

输入

每个测试用例由多行组成，第一行是两个整数P和R，P代表供水点数（1<=P<=50），每个点都对应1到P中的一个唯一编号。R代表可能的铺设路径数，路径数可能有非常多。接下有R行，每行格式如下：

节点A编号 节点B编号 路径成本

路径成本不超过100。

测试用例之间有一空行分开。输入结束用P=0表示，注意没有R值。

输出

每个测试用例占用一行输出最低总成本。

样例输入 

1 0

2 3
1 2 37
2 1 17
1 2 68

3 7
1 2 19
2 3 11
3 1 7
1 3 5
2 3 89
3 1 91
1 2 32

5 7
1 2 5
2 3 7
2 4 8
4 5 11
3 5 10
1 5 6
4 2 12

0

样例输出 

0
17
16
26

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
 
const int MAX_P = 50;
 
int P, R, cost;
int fa[MAX_P+1]; // 并查集数组
 
struct Edge {
    int u, v, w;
    Edge(int u_, int v_, int w_) : u(u_), v(v_), w(w_) {}
    bool operator<(const Edge& e) const {
        return w < e.w;
    }
};
 
vector<Edge> edges;
 
int find(int x) {
    if (x == fa[x]) return x;
    return fa[x] = find(fa[x]);
}
 
int main() {
    while (cin >> P && P != 0) {
        cin >> R;
        edges.clear();
        for (int i = 1; i <= P; i++) fa[i] = i; // 初始化并查集
        for (int i = 0; i < R; i++) {
            int u, v, w;
            cin >> u >> v >> w;
            edges.push_back(Edge(u, v, w));
        }
        sort(edges.begin(), edges.end()); // 对边权排序
        cost = 0;
        for (int i = 0; i < R; i++) {
            int u = edges[i].u, v = edges[i].v, w = edges[i].w;
            int root_u = find(u), root_v = find(v);
            if (root_u != root_v) { // 若 u, v 不连通则将边 (u, v) 加入最小生成树
                cost += w;
                fa[root_u] = root_v;
            }
        }
        cout << cost << endl;
    }
    return 0;
}