其他分享
首页 > 其他分享> > Leetcode 1020. 飞地的数量

Leetcode 1020. 飞地的数量

作者:互联网

Leetcode 1020. 飞地的数量

题目

给出一个二维数组 A,每个单元格为 0(代表海)或 1(代表陆地)。

移动是指在陆地上从一个地方走到另一个地方(朝四个方向之一)或离开网格的边界。

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

示例 1:

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

示例 2:

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

思路

代码

class Solution {
public:
    void dfs(vector<vector<int>>& A, int row, int col) {
        if (row < 0 || row >= A.size() || col < 0 || col >= A[0].size() || A[row][col] == 0) return;

        A[row][col] = 0;
        dfs(A, row, col - 1);
        dfs(A, row, col + 1);
        dfs(A, row - 1, col);
        dfs(A, row + 1, col);
    }

    int numEnclaves(vector<vector<int>>& A) {
        for (int row = 0;row < A.size();++row) {
            if (A[row][0] == 1) dfs(A, row, 0);

            if (A[row][A[0].size() - 1] == 1) dfs(A, row, A[0].size() - 1);
        }

        for (int col = 0;col < A[0].size();++col) {
            if (A[0][col] == 1) dfs(A, 0, col);

            if (A[A.size() - 1][col] == 1) dfs(A, A.size() - 1, col);
        }

        int res = 0;

        for (auto& rowline:A) {
            for (auto& data:rowline) {
                if (data == 1) ++res;
            }
        }

        return res;
    }
};

标签:陆地,1020,飞地,dfs,Leetcode,int,size,col,row
来源: https://blog.csdn.net/weixin_43891775/article/details/111655317