Leetcode之深度优先搜索(DFS)专题-1020. 飞地的数量(Number of Enclaves)
深度优先搜索的解题详细介绍,点击
给出一个二维数组 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
- 所有行的大小都相同
思路与130题一样。
步骤如下:
- 找边框上的1,因为只有这些1,并且与边框上的1相连的1,才可以到达陆地。
- 把找到的这些1变成0
- 重新遍历,统计1的个数并返回。
AC代码:
class Solution {
int dirx[] = {0,0,1,-1};
int diry[] = {1,-1,0,0};
public int numEnclaves(int[][] A) {
if(A.length == 0 || A == null) return 0;
int ans = 0;
int vis[][] = new int[A.length][A[0].length];
for(int i=0;i<A.length;i++){
for(int j=0;j<A[0].length;j++){
boolean flag = i==0||i==A.length-1||j==0||j==A[0].length-1;
if(flag && A[i][j]==1){
change(A,i,j);
}
}
}
for(int i=0;i<A.length;i++){
for(int j=0;j<A[0].length;j++){
if(A[i][j]==1)
ans++;
}
}
return ans;
} public void change(int[][] A, int x, int y) {
A[x][y] = 0;
for (int i = 0; i < 4; i++) {
int xx = x + dirx[i];
int yy = y + diry[i];
if (xx >= 0 && xx < A.length && yy >= 0 && yy < A[0].length
&& A[xx][yy] == 1) {
change(A, xx, yy);
}
}
}
}