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

提示：

1 <= A.length <= 500
1 <= A[i].length <= 500
0 <= A[i][j] <= 1
所有行的大小都相同

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

 

 

解题思路

class Solution {
public:
    queue<pair<int,int>> que;
    int numEnclaves(vector<vector<int>>& A) {
        int ans = 0,flag = 0,num = 0;
        int row = A.size(),col = A[0].size();
        int dx[] = {-1,0,1,0},dy[] = {0,-1,0,1};
        for(int i=0;i<row;++i){
            for(int j=0;j<col;++j){
                if(A[i][j] == 1){
                    num = 1;
                    flag = 0;
                    A[i][j] = 0;
                    que.push(make_pair(i,j));
                    while(!que.empty()){
                        int x = que.front().first;
                        int y = que.front().second;
                        que.pop();
                        for(int k=0;k<4;++k){
                            int tx = x + dx[k];
                            int ty = y + dy[k];
                            if(tx>=0&&tx<row&&ty>=0&&ty<col&&A[tx][ty]==1){
                                A[tx][ty] = 0;
                                que.push(make_pair(tx,ty));
                                ++num;
                            }else if(tx<0||tx>=row||ty<0||ty>=col) flag = 1;
                        }
                    }
                    if(flag == 0) ans += num;
                }
            }
        }
        return ans;
    }
};