【LeetCode】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

思路

既然与边界相连的方格都不是飞地，那么先对边界的所有格子做dfs搜索，剩下没有被搜索到的格子都是飞地。

代码

class Solution {
    
    
private:
    const int dx[4] = {
    
    0, 0, 1, -1};
    const int dy[4] = {
    
    -1, 1, 0, 0};
    void dfs(vector<vector<int>>& grid, int x, int y) {
    
    
        if (x < 0 || x >= grid.size() 
            || y < 0 || y >= grid[0].size()
            || !grid[x][y]) return;
        grid[x][y] = 0;
        for (int i = 0; i < 4; ++i) {
    
    
            int nx = x + dx[i], ny = y + dy[i];
            dfs(grid, nx, ny);
        }  
    }

public:
    int numEnclaves(vector<vector<int>>& grid) {
    
    
        int ans = 0;
        int n = grid.size();
        int m = grid[0].size();
        for (int i = 0; i < n; ++i) {
    
    
            dfs(grid, i, 0);
            dfs(grid, i, m - 1);
        }
        for (int i = 0; i < m; ++i) {
    
    
            dfs(grid, 0, i);
            dfs(grid, n - 1, i);
        }
        for (int i = 0; i < grid.size(); ++i) {
    
    
            for (int j = 0; j < grid[0].size(); ++j) {
    
    
                if (grid[i][j]) ans++;
            }
        }
        return ans;
    }
};