蓝桥杯——合根植物

题目

问题描述
　　w星球的一个种植园，被分成 m * n 个小格子（东西方向m行，南北方向n列）。每个格子里种了一株合根植物。
　　这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。
　　如果我们告诉你哪些小格子间出现了连根现象，你能说出这个园中一共有多少株合根植物吗？

Input

第一行，两个整数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

Output

样例输出
5
样例说明
其合根情况参考下图

思路

利用并查集来解决，每个点都把他们的父节点初始赋值是他们自己，等于是每个植物都是一个集合，然后在接下来的k组输入里面，先寻找他们的父节点，如果一致，则继续，如果不一致，则需要合并其中一个父节点为另外一个的父节点【因为其实有一个的父节点是它自己，其实也就是把其中一个作为另外一个的父节点】，最终循环遍历一遍，只要寻找出多少个父节点是它本身的节点就是有多少个集合。

代码

import java.util.Scanner;
public class Main {
    static int father[]=new int[1000005];//即每个结点的所属的父节点
    public static void main(String[] args) {
        Scanner reader=new Scanner(System.in);
        int m=reader.nextInt();//行
        int n=reader.nextInt();//列

        int length=m*n;
        //一开始，每个人的初始父节点都是自己，即每个人都自称一个集合
        for (int i=1;i<=length;++i)
            father[i]=i;

        int k=reader.nextInt();//k组输入
        for (int i=0;i<k;++i)
        {
            //接收同根的两个植物
            int a=reader.nextInt();
            int b=reader.nextInt();
            //判断两个点的父节点是否相等，如果不相等
            if (findfather(a)!=findfather(b))
            {
                union(a,b);
            }
        }

        int ans=0;
        for (int i=1;i<=length;++i)
            if (father[i]==i)
                ++ans;
        System.out.println(ans);

    }

    private static void union(int a, int b) {
        a=findfather(a);
        b=findfather(b);
        //两个不相等，加到同一个队伍里
        if (a!=b)
        {
            father[b]=a;
        }
    }

    private static int findfather(int a)
    {
        //如果a的父节点是自己，那么表明它是单个植物
        if (a==father[a])
            return a;
        //如果该植物的父节点不是自己，那么表明我们要它的父节点，找到最上层的父节点，即进行递归寻找到最上层的父节点
        father[a]=findfather(father[a]);
        return father[a];
    }
}

结果

算是学习了另外一种简答算法的一点点皮毛，还挺好的，感觉很神奇。