【CCF】并查集 - 模板例题

问题描述

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

---

题解

直接套模板即可

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
import java.util.Arrays;
import java.util.Comparator;

public class csp_2014_12_4 {

    public static void main(String[] args) throws IOException {
        StreamTokenizer in = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
        
        in.nextToken();
        int n = (int) in.nval;
        in.nextToken();
        int m = (int) in.nval;
        
        Tri[] tris = new Tri[m];
        int num = 0;
        while (num != m) {
            in.nextToken();
            int u = (int) in.nval;
            in.nextToken();
            int v = (int) in.nval;
            in.nextToken();
            int w = (int) in.nval;
            tris[num ++] = new Tri(u, v, w);
        }
        
        Comparator cmp = new Comparator() {
            public int compare(Tri t1, Tri t2) {
                return t1.val - t2.val;
            }
        };
        Arrays.sort(tris, 0, m, cmp);
        
        Unionx uset = new Unionx(n);
        int cost = 0;
        num = 0;
        for (int i = 0; i < m; i ++) {
            int x = uset.find(tris[i].begin);
            int y = uset.find(tris[i].end);
            if (x != y) {
                num ++;
                uset.unionx(x, y);
                cost += tris[i].val;
            }
            if (num == n) {
                break;
            }
        }
        System.out.println(cost);
    }
    
    static class Unionx {
        int[] f;
        public Unionx(int n) {
            this.f = new int[n + 5];
            for (int i = 1; i <= n; i ++) {
                f[i] = i;
            }
        }
        
        public int find(int x) {
            return f[x] == x ? x : find(f[x]);
        }
        
        public void unionx(int x, int y) {
            x = find(x);
            y = find(y);
            if (x != y) {
                f[x] = y;
            }
        }
    }
    
    static class Tri {
        int begin;
        int end;
        int val;
        public Tri(int begin, int end, int val) {
            this.begin = begin;
            this.end = end;
            this.val = val;
        }
    }

}

更多CCF题目代码可参考：https://github.com/dhwgithub/CCF-CSP-Recording