数据结构与算法(Java) 41：岛问题

题目 一个矩阵中只有0和1两种值，每个位置都可以和自己的上、下、左、右 四个位置相连，如果有一片1连在一起，这个部分叫做一个岛，求一个 矩阵中有多少个岛？

举例： 0 0 1 0 1 0

            1 1 1 0 1 0

            1 0 0 1 0 0

            0 0 0 0 0 0

这个矩阵中有三个岛。

思路 遍历矩阵，若当前数值为1，则以当前元素为感染源，进行感染，并计数+1。感染过程即：递归查询当前元素的上下左右是否为1，若为1，则将数值改为2，否则直接返回。

这是对于小型数据，单CPU来说可以这样做，但是如果矩阵特别特别大，这样遍历就会及其耗时。这时候就需要多台机器来操作，并且使用到并查集结构。主要应用分治的思想。若有m台机器，则把矩阵分为m块，每台机器一块小矩阵。对于每台机器来说，用上述简单方法计算出岛的个数，并记录小矩阵边缘的节点，边缘节点属于一个岛的用相同的符号标记，属于不同岛的用不同符号标记。最后在进行合并时，首先将所有机器计算所得岛的个数加到一起，然后对于每个分割边缘，例如：左[1]="A", 左[2]="A", 左[4]="B", 右[2] ="C", 右[3]="C", 右[4]="D"。遍历边缘两边几点，若左右两边都有元素，则用并查集观察二者是否来自同一个集合，若来自同一集合，则跳过进行下面的观察，否则将两边元素所在的集合用并查集进行合并，并将岛的个数-1.

package algorithm.section6;

public class Islands {
    public static int countOfLands(int[][] matrix){
        if (matrix.length == 0) return 0;

        int count = 0;
        for (int i = 0; i < matrix.length; i++){
            for (int j = 0; j < matrix[0].length; j++){
                if (matrix[i][j] == 1){
                    count++;
                    infect(matrix, i, j);
                }
            }
        }

        return count;
    }

    public static void infect(int[][] matrix, int i, int j){
        if (i < 0 || i > matrix.length || j < 0 || j > matrix[0].length || matrix[i][j] != 1) return;
        if (matrix[i][j] == 1) matrix[i][j] = 2;

        infect(matrix, i - 1, j);
        infect(matrix, i + 1, j);
        infect(matrix, i, j - 1);
        infect(matrix, i, j + 1);
    }

    public static void main(String[] args){
        int[][] m1 = {  { 0, 0, 0, 0, 0, 0, 0, 0, 0 },
                        { 0, 1, 1, 1, 0, 1, 1, 1, 0 },
                        { 0, 1, 1, 1, 0, 0, 0, 1, 0 },
                        { 0, 1, 1, 0, 0, 0, 0, 0, 0 },
                        { 0, 0, 0, 0, 0, 1, 1, 0, 0 },
                        { 0, 0, 0, 0, 1, 1, 1, 0, 0 },
                        { 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
        System.out.println(countOfLands(m1));

        int[][] m2 = {  { 0, 0, 0, 0, 0, 0, 0, 0, 0 },
                        { 0, 1, 1, 1, 1, 1, 1, 1, 0 },
                        { 0, 1, 1, 1, 0, 0, 0, 1, 0 },
                        { 0, 1, 1, 0, 0, 0, 1, 1, 0 },
                        { 0, 0, 0, 0, 0, 1, 1, 0, 0 },
                        { 0, 0, 0, 0, 1, 1, 1, 0, 0 },
                        { 0, 0, 0, 0, 0, 0, 0, 0, 0 }, };
        System.out.println(countOfLands(m2));
    }
}