CCF201412-4 最优灌溉（解法二）（100分）

试题编号：	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。

问题链接：CCF201412试题。

问题描述：（参见上文）。

问题分析：这是一个最小生成树的为问题，解决的算法有Kruskal（克鲁斯卡尔）算法和Prim（普里姆） 算法。

程序说明：本程序使用Kruskal算法实现。有关最小生成树的问题,使用克鲁斯卡尔算法更具有优势，只需要对所有的边进行排序后处理一遍即可。程序中使用了并查集，用来判定加入一条边后会不会产生循环。n个结点的图，其最小生成树应该是n-1条边，这个作为程序处理的结束条件。这个程序实现Kruskal算法部分的逻辑和代码都是否简洁易懂。程序中，图采用边列表的方式存储，按边的权从小到大顺序放在优先队列中，省去了排序。

提交后得100分的C++语言程序如下：

/* CCF201412-4 最优灌溉 */

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

// 并查集类
class UF {
private:
    vector<int> v;
public:
    UF(int n) {
        for(int i=0; i<=n; i++)
            v.push_back(i);
    }

    int Find(int x) {
        for(;;) {
            if(v[x] != x)
                x = v[x];
            else
                return x;
        }
    }

    bool Union(int x, int y) {
        x = Find(x);
        y = Find(y);
        if(x == y)
            return false;
        else {
            v[x] = y;
            return true;
        }
    }
};

struct edge {
    int src, dest, cost;
    bool operator < (const edge& n) const {
        return cost > n.cost;
    }
};

int main()
{
    priority_queue<edge> q;     // 优先队列，用于存储边列表
    edge e;

    // 输入数据
    int n, m;
    cin >> n >> m;
    for(int i=1; i<=m; i++) {
        cin >> e.src >> e.dest >> e.cost;
        q.push(e);
    }

    // Kruskal算法
    UF uf(n);
    int ans=0, count=0;
    for(;;) {
        e = q.top();
        q.pop();

        if(uf.Find(e.src) != uf.Find(e.dest)) {
            uf.Union(e.src, e.dest);

            ans += e.cost;

            if(++count == n -1)
                break;
        }
    }

    // 输出结果
    cout << ans << endl;

    return 0;
}

/*
测试数据与结果两组：
6 10
1 2 3
1 3 1
1 4 6
2 3 5
2 5 3
3 4 5
3 5 6
3 6 4
4 6 2
5 6 6
13

4 4
1 2 1
2 3 4
2 4 2
3 4 3
6
*/