ccf_201412-4_最优灌溉

问题描述
试题编号： 201412-4
试题名称： 最优灌溉
时间限制： 1.0s
内存限制： 256.0MB
问题描述：
　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。
输入格式
　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。
输出格式
　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。
样例输入
4 4
1 2 1
2 3 4
2 4 2
3 4 3
样例输出
6
样例说明
　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。
评测用例规模与约定
　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤ci≤10,000。

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

struct Node{
    int a,b,c;
    Node() = default;
    Node(int a,int b,int c):a(a),b(b),c(c){}
    bool operator <(const Node & n) const {
        return c > n.c;
    }
};
class UF{
public:
    vector<int> id,sz;
    UF() = default;
    explicit UF(int n){
        for (int i = 0; i <= n; ++i) {
            id.push_back(i);
            sz.push_back(1);
        }
    }
    int find_(int p){
        while (p != id[p]) p = id[p];
        return p;
    }
    bool connect(int p,int q){
        return find_(p) == find_(q);
    }
    void union_(int p, int q){
        int i = find_(p);
        int j = find_(q);
        if(i == j) return;
        if(sz[i] < sz[j]){
            id[i] = j;
            sz[j] += sz[i];
        } else{
            id[j] = i;
            sz[i] += sz[j];
        }
    }
};
int main(){
    int m,n,cost = 0;
    priority_queue<Node> node;
    cin>>n>>m;
    UF uf(n);
    for (int i = 0; i < m; ++i) {
        int a,b,c;
        cin>>a>>b>>c;
        node.emplace(a,b,c);
    }
    for (int i = 0; !node.empty() && i < n-1 ; ) {
       Node t = node.top();
       node.pop();
       if(!uf.connect(t.a,t.b)){
           cost += t.c;
           uf.union_(t.a,t.b);
           ++i;
       }
    }
    cout<<cost;
}