每日OJ题_多源BFS②_力扣1020. 飞地的数量

目录

力扣1020. 飞地的数量

解析代码

力扣1020. 飞地的数量

1020. 飞地的数量

难度 中等

给你一个大小为 m x n 的二进制矩阵 grid ，其中 0 表示一个海洋单元格、1 表示一个陆地单元格。

一次 移动 是指从一个陆地单元格走到另一个相邻（上、下、左、右）的陆地单元格或跨过 grid 的边界。

返回网格中 无法 在任意次数的移动中离开网格边界的陆地单元格的数量。

示例 1：

输入：grid = [[0,0,0,0],[1,0,1,0],[0,1,1,0],[0,0,0,0]]
输出：3
解释：有三个 1 被 0 包围。一个 1 没有被包围，因为它在边界上。

示例 2：​​​​​​​

输入：grid = [[0,1,1,0],[0,0,1,0],[0,0,1,0],[0,0,0,0]]
输出：0
解释：所有 1 都在边界上或可以到达边界。

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 500
• grid[i][j] 的值为 0 或 1

class Solution {
public:
    int numEnclaves(vector<vector<int>>& grid) {
        
    }
};

解析代码

        正难则反，从边上的 1 开始搜索，把与边上 1 相连的联通区域全部标记一下，然后再遍历一遍矩阵，看看哪些位置的 1 没有被标记即可标记的时候，可以用多源BFS解决。

class Solution {
    int dx[4] = {0, 0, -1, 1};
    int dy[4] = {1, -1, 0, 0};

public:
    int numEnclaves(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        queue<pair<int, int>> q;
        for(int j = 0; j < n; ++j) // 1. 处理边界的1
        {
            if(grid[0][j] == 1) // 第一行
            {
                grid[0][j] = -1;
                q.push({0, j});
            }
            if(grid[m - 1][j] == 1) // 最后一行
            {
                grid[m - 1][j] = -1;
                q.push({m - 1, j});
            }
        }
        for(int i = 0; i < m; ++i)
        {
            if(grid[i][0] == 1) // 第一列
            {
                grid[i][0] = -1;
                q.push({i, 0});
            }
            if(grid[i][n - 1] == 1) // 最后一列
            {
                grid[i][n - 1] = -1;
                q.push({i, n - 1});
            }
        }

        while(!q.empty())
        {
            auto[a, b] = q.front();
            q.pop();
            for(int i = 0; i < 4; ++i)
            {
                int x = a + dx[i], y = b + dy[i];
                if(x >= 0 && x < m && y >= 0 && y < n && grid[x][y] == 1)
                {
                    grid[x][y] = -1;
                    q.push({x, y});
                }
            }
        }

        int ret = 0;
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(grid[i][j] == 1)
                    ++ret;
            }
        }
        return ret;
    }
};