HUSTOJ 基础题组1 问题 O: 畅通工程

题目描述

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

输入

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

输出

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

样例输入 Copy

3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

样例输出 Copy

3
?

#include<iostream>
#include<algorithm>
using namespace std;
 
int set[10024] = { 0 };
struct city {
    int start;
    int end;
    int distance;
}c[10002];
 
bool cmp(city c1, city c2) {
    return c1.distance < c2.distance;
}
int Where_are_you_from(int pos) {
    while (pos != set[pos]) {
        pos = set[pos];
    }
    return pos;
}
bool no_loop(int a, int b) {
    a = Where_are_you_from(a);
    b = Where_are_you_from(b);
    if (a == b)
        return 0;
    return 1;
}
void merge(int start, int end) {
    int a = Where_are_you_from(start);
    int b = Where_are_you_from(end);
    if (a < b)
        set[b] = a;
    else
        set[a] = b;
}
bool is_tree(int city_numbers) {
    int a = set[1];
    for (int i = 1; i <= city_numbers;i++) {
        if (Where_are_you_from(i) != a)
            return 0;
    }
    return 1;
}
 
int main() {
    int road_numbers, city_numbers;
    while (cin >> road_numbers >> city_numbers) {
        if (road_numbers == 0)
            break;
 
        //读取每一个可能的道路
        for (int i = 0; i < road_numbers; i++) {
            cin >> c[i].start >> c[i].end >> c[i].distance;
            if (c[i].start > c[i].end) {
                int temp = c[i].start;
                c[i].start = c[i].end;
                c[i].end = temp;
            }
        }
 
        //初始化并查集
        for (int i = 1; i <= city_numbers; i++) {
            set[i] = i;
        }
 
        //排序
        sort(c, c + road_numbers, cmp);
 
 
        int result = 0;
        for (int i = 0; i < road_numbers; i++) {
            int start = c[i].start;
            int end = c[i].end;
            if (no_loop(start,end)) {
                merge(start,end);
                result += c[i].distance;
            }
        }
 
        if (is_tree(city_numbers))
            cout << result << endl;
        else
            cout << "?" << endl;
    }
}