蓝桥杯真题：合根植物（并查集）JAVA实现

题目描述

ww 星球的一个种植园，被分成 m \times nm×n 个小格子（东西方向 mm 行，南北方向 nn 列）。每个格子里种了一株合根植物。

这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。

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

输入描述
第一行，两个整数 m,nm,n，用空格分开，表示格子的行数、列数（1 \leq m,n \leq 10001≤m,n≤1000）。

接下来一行，一个整数 kk (0 \leq k \leq 10^50≤k≤10
5
)，表示下面还有 kk 行数据。

接下来 kk 行，第行两个整数 a，ba，b，表示编号为 aa 的小格子和编号为 bb 的小格子合根了。

格子的编号一行一行，从上到下，从左到右编号。

比如：5 \times 45×4 的小格子，编号：

1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20
copy
输出描述
输出植物数量。

输入输出样例
示例
输入

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
copy
输出

5
copy
样例说明

其合根情况参考下图：

运行限制
最大运行时间：2s
最大运行内存: 256M
非常简单的一道题，很适合我这种没有信心的选手做。

代码如下：

public class Main {
    public static void main(String[] args) {
        Scanner sc=new Scanner(System.in);
        int M=sc.nextInt(),N=sc.nextInt();
        UnionFindDemo02 uf=new UnionFindDemo02(M*N);//M行N列，所以初始化时有MxN朵花
        int numbers=sc.nextInt();//获取接下来数据的行数
        for (int i = 0; i < numbers; i++) {
            int a=sc.nextInt();int b=sc.nextInt();
            uf.union(a,b);//将两朵花连根
        }
        System.out.println(uf.count());//获取花的数量
    }
}
class UnionFindDemo02 {//实现并查集
    private int[] id;//花的id
    private int count;//花的数量

    public UnionFindDemo02(int N) {
        count = N;//刚刚初始化时，默认每朵花都是不连根的，因此有n朵花
        id = new int[N];
        //初始化时，每朵花都是不连根的，因此每个花的id都为本身的id
        for (int i = 0; i < id.length; i++) {
            id[i] = i;
        }
    }
	//返回花的数量
    public int count() {
        return count;
    }
	//判断两朵花是否连根
    public boolean connected(int p, int q) {
        return find(p) == find(q);
    }
	//寻找p花根节点的id
    public int find(int p) {
       while(p!=id[p]) p=id[p];
       return p;
    }
    //将编号为p的花和编号为q的花连根
    public void union(int p,int q){
        int pID=find(p);//p花根节点的id
        int qID=find(q);//q花根节点的id
        if(pID==qID) return;//如果两朵花根节点id相同，那么他们是连根的，直接返回即刻
        else{
            id[pID]=qID;//让p花的根节点指向q花根节点，两花合并
            count--;//两朵花合并后，花的总数减一
        }
    }
}