给出一个二维数组 A,每个单元格为 0(代表海)或 1(代表陆地)。
移动是指在陆地上从一个地方走到另一个地方(朝四个方向之一)或离开网格的边界。
返回网格中无法在任意次数的移动中离开网格边界的陆地单元格的数量。
示例 1:
输入:[[0,0,0,0],[1,0,1,0],[0,1,1,0],[0,0,0,0]]
输出:3
解释:
有三个 1 被 0 包围。一个 1 没有被包围,因为它在边界上。
解题思路
从网格边界出发,像网格内部进行深度优先搜索,将可到达的地方全部置为水域,最后遍历矩阵,找出无法到达的陆地。
代码
class Solution {public int numEnclaves(int[][] A) {int[][] dir=new int[][]{{-1,0},{1,0},{0,1},{0,-1}};for(int i=0;i<A.length;i++)//遍历边界{Enclaves(A, dir,i,0);Enclaves(A, dir,i,A[0].length-1);}for(int i=0;i<A[0].length;i++){Enclaves(A, dir,0,i);Enclaves(A, dir,A.length-1,i);}int ans=0;for(int i=1;i<A.length-1;i++)for(int j=1;j<A[0].length-1;j++)if(A[i][j]==1) ans++;return ans;}public void Enclaves(int[][] A,int[][] dir,int x,int y) {if(x<0||y>=A[0].length||y<0||x>=A.length)return ;if(A[x][y]==0) return ;A[x][y]=0;for(int[] d:dir){int nextX=d[0]+x,nextY=d[1]+y;Enclaves(A, dir, nextX, nextY);}}
}