2017年javaC组决赛第6题 合根植物

题目：合根植物

w星球的一个种植园，被分成 m * n 个小格子（东西方向m行，南北方向n列）。每个格子里种了一株合根植物。
这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。

如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？

输入格式：
第一行，两个整数m，n，用空格分开，表示格子的行数、列数（1<m,n<1000）。
接下来一行，一个整数k，表示下面还有k行数据(0<k<100000)
接下来k行，第行两个整数a，b，表示编号为a的小格子和编号为b的小格子合根了。

格子的编号一行一行，从上到下，从左到右编号。
比如：5 * 4 的小格子，编号：
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20

样例输入：
5 4
16
2 3
1 5
5 9
4 8
7 8
9 10
10 11
11 12
10 14
12 16
14 18
17 18
15 19
19 20
9 13
13 17

样例输出：
5

其合根情况参考图

资源约定：
峰值内存消耗 < 256M
CPU消耗 < 2000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意：主类的名字必须是：Main，否则按无效代码处理。

思路：并查集

如果知道并查集，并不难想到使用并查集。

并查集有两个基本操作，查询和合并

• 查找元素所在的集合，即根节点。
• 将两个元素所在的集合合并为一个集合。

算法：在初始化的时候，设置一个值表示总共有多少种植物。每输入两个数时，先检查两个数是不是在同一个集合，如果不在则将两个数所在的集合合并，且植物种数减一。

public class Main {
    public static class UN{
        int[] un;
        int[] r;
        int cnt; // 有多少种植物
        public UN(int n) {
            cnt = n;
            un = new int[n + 1];
            for(int i = 1; i <= n; i++)
                un[i] = i;
            r = new int[n + 1];
        }

        public void union(int a, int b) {
            int af = f(a);
            int bf = f(b);

            if(af != bf) {
                cnt--;

                if(r[af] > r[bf]){
                    un[bf] = un[af];
                }
                else{
                    un[af] = un[bf];
                    r[bf] += r[af] == r[bf] ? 1 : 0;
                }
            }
        }

        public int f(int x) {
            if(x == un[x]) return x;
            return un[x] = f(un[x]);
        }

        public int cnt() {
            return cnt;
        }
    }


    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        int n = sc.nextInt();
        int k = sc.nextInt();

        int ind = 1;
        UN un = new UN(m * n);

        for(int i = 0; i < k; i++) {
            int a = sc.nextInt();
            int b = sc.nextInt();

            un.union(a, b);
        }

        System.out.println(un.cnt());
    }
}