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的格子进行深度遍历,全部标0,最后自然剩下无法到达边界的格子。
打个比方,就好像边界上的陆地会伸出触手破坏所有它够得到的陆地,最终只有被海包围的陆地能幸存
class Solution {
public int numEnclaves(int[][] A) {
if(A == null)
return 0;
int row = A.length;
int col = A[0].length;
int res = 0;
if(row == 0 || col == 0)
return 0;
for(int i = 0; i < row; i++){
dfs(i, 0, A);
dfs(i, col - 1, A);
}
for(int j = 0; j < col; j++){
dfs(0, j, A);
dfs(row - 1, j, A);
}
for(int i = 0; i < row; i++){
for(int j = 0; j < col; j++){
res += A[i][j];
}
}
return res;
}
public void dfs(int i, int j, int[][] A){
if(i < 0 || j < 0 || i >= A.length || j >= A[0].length)
return ;
if(A[i][j] == 0)
return ;
A[i][j] = 0;
dfs(i + 1, j, A);
dfs(i - 1, j, A);
dfs(i, j + 1, A);
dfs(i, j - 1, A);
}
}