Leetcode 5791：统计子岛屿

题目描述

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

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

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

 

示例 1：

输入：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 个子岛屿。
示例 2：

输入：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 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-sub-islands
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    int m;
    int n;
    vector<int> dx = {-1,0,1,0};
    vector<int> dy = {0,1,0,-1};
    void dfs(vector<vector<int>>& grid, int x, int y, vector<pair<int, int>> &tmp){
        grid[x][y] = 0;
        tmp.push_back(make_pair(x, y));
        for(int i = 0; i < 4; i++){
            int tx = x+dx[i];
            int ty = y+dy[i];
            if(tx >= 0 && tx < m && ty >=0 && ty < n && grid[tx][ty] == 1){
                dfs(grid, tx, ty, tmp);
            }
        }
    }
    int countSubIslands(vector<vector<int>>& grid1, vector<vector<int>>& grid2) {
        if(grid1.size() == 0 || grid1[0].size() == 0) return 0;
        m = grid1.size();
        n = grid1[0].size();
        int ans = 0;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(grid2[i][j] == 1){
                    vector<pair<int, int>> tmp;
                    dfs(grid2, i, j, tmp);
                    bool flag = true;
                    for(auto &it : tmp){
                        if(grid1[it.first][it.second] == 0){
                            flag = false;
                            break;
                        }
                    }
                    if(flag) ans++;
                }
            }
        }
        return ans;
    }
};