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 都在边界上或可以到达边界。

思路

• 我们可以从四周边界考虑， 反向思维， 从四周的陆地可以走到哪些陆地
• 我们可以使用DFS一次走完一条路径上的所有的陆地， 将可以走到的陆地置为0
• 最后统计网格中有多少个1即可， 因为能走到的1肯定已经被修改成了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;
    }
};