统计子岛屿的数量

统计子岛屿

题目描述

给你两个 m x n 的二进制矩阵 grid1 和 grid2 ,它们只包含 0 (表示水域)和 1 (表示陆地)。一个 岛屿 是由 四个方向 (水平或者竖直)上相邻的 1 组成的区域。任何矩阵以外的区域都视为水域。

如果 grid2 的一个岛屿,被 grid1 的一个岛屿 完全 包含,也就是说 grid2 中该岛屿的每一个格子都被 grid1 中同一个岛屿完全包含,那么我们称 grid2 中的这个岛屿为 子岛屿 。

请你返回 grid2 中 子岛屿 的 数目 。

统计子岛屿的数量_第1张图片
统计子岛屿的数量_第2张图片

思路:

对于(i, j)来说,有四种情况

case1:grid1[i][j] = 1, grid2[i][j] = 1.

case2:grid1[i][j] = 1, grid2[i][j] = 0.

case3: grid1[i][j] = 0, grid2[i][j] = 1.

case4: grid1[i][j] = 0, grid2[i][j] = 0.

对于case4,我们完全不用关心;剩下的case1、case2、case3我们再看,由于是grid1包含grid2,对于case3,如果grid2是陆地,grid1是海水,那么grid1就不包含grid2,那么我们就可以提前干掉grid2中的点;对于case2,如果grid1是陆地,grid2是海水,那么grid2必然不是grid1的子岛,我们可以不做任何操作;对于case1,那么grid2必然是grid1的子岛,我们就计数,然后dfs。

下面看代码:

	public int countSubIslands(int[][] grid1, int[][] grid2) {
        int m = grid1.length, n = grid1[0].length, count = 0;
        // 先排除不是子岛屿的节点
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
                if(grid2[i][j] == 1 && grid1[i][j] == 0){
                    dfs(grid2, i, j, m, n);
                }
            }
        }
        System.out.println("======");
        for(int i = 0;i < m;i++){
            for(int j = 0;j < n;j++){
            	// 对于条件case1和case2来说
            	// 这块可以优化成if(grid2[i][j] == 1)
                if(grid2[i][j] == 1 && grid1[i][j] == 1){
                    count++;
                    dfs(grid2, i, j, m, n);
                }
            }
        }
        return count;
    }

    public void dfs(int[][] grid, int i, int j, int m, int n){
        if(i >= m || i < 0 || j >= n || j < 0
                || grid[i][j] == 0 ){
            return;
        }
        grid[i][j] = 0;
        dfs(grid, i + 1, j, m, n);
        dfs(grid, i - 1, j, m, n);
        dfs(grid, i, j + 1, m, n);
        dfs(grid, i, j - 1, m, n);
    }

    public static void main(String[] args) {
        int[][] grid1 = {{1,1,1,0,0},{0,1,1,1,1},{0,0,0,0,0},{1,0,0,0,0},{1,1,0,1,1}};
        int[][] grid2 = {{1,1,1,0,0},{0,0,1,1,1},{0,1,0,0,0},{1,0,1,1,0},{0,1,0,1,0}};
        CountSubIslands countSubIslands = new CountSubIslands();
        countSubIslands.countSubIslands(grid1, grid2);

    }

你可能感兴趣的:(回溯,java)