LeetCode刷题笔记 ~ 5791. 统计子岛屿

这是我参与更文挑战的第6天，活动详情查看：更文挑战

5791. 统计子岛屿

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

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

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

示例：

输入：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]], 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]]
输出：3
解释：如上图所示，左边为 grid1 ，右边为 grid2 。
grid2 中标红的 1 区域是子岛屿，总共有 3 个子岛屿。

输入：grid1 = [[1,0,1,0,1],[1,1,1,1,1],[0,0,0,0,0],[1,1,1,1,1],[1,0,1,0,1]], grid2 = [[0,0,0,0,0],[1,1,1,1,1],[0,1,0,1,0],[0,1,0,1,0],[1,0,0,0,1]]
输出：2 
解释：如上图所示，左边为 grid1 ，右边为 grid2 。
grid2 中标红的 1 区域是子岛屿，总共有 2 个子岛屿。

提示

• m == grid1.length == grid2.length
• n == grid1[i].length == grid2[i].length
• 1 <= m, n <= 500
• grid1[i][j] 和 grid2[i][j] 都要么是 0 要么是 1 。

个人思路解析

方式一：DFS

1. 遍历grid2，找到每一个小岛的位置
2. DFS渠道子岛屿的范围
3. 与grid1进行比较，查看是否被完全包围

class Solution {
    public int countSubIslands(int[][] grid1, int[][] grid2) {
        int ans = 0;
        for(int i = 0; i < grid2.length; ++i){
            for(int j = 0; j < grid2[0].length; ++j){
                // 找到 grid2 中每一个岛屿，并确定是否被 grid1 所包围
                if(grid2[i][j] == 1 && dfs(grid1, grid2, i, j)){
                    ++ans;
                }
            }
        }
        return ans;
    }

    public boolean dfs(int[][] grid1, int[][] grid2, int m, int n){
        // 边界判断
        if(m < 0 || m >= grid2.length || n < 0 || n >= grid2[0].length || grid2[m][n] == 0){
            return true;
        }

        // 超过 grid1 的范围，直接返回
        if(grid1[m][n] == 0){
            return false;
        }

        // 遍历过的岛屿修改为 0，防止溢出
        grid2[m][n] = 0;

        // 分别向上下左右延申，搜索岛屿的范围
        // 这里用 &，只要有一处岛屿没被包含，则该岛屿不为子岛屿
        boolean flag = true;
        flag &= dfs(grid1, grid2, m + 1, n);
        flag &= dfs(grid1, grid2, m - 1, n);
        flag &= dfs(grid1, grid2, m, n + 1);
        flag &= dfs(grid1, grid2, m, n - 1);
        
        // 返回结果
        return flag;
    }
}

提交结果

题目来源力扣：leetcode-cn.com/problems/co…