题目描述
给你一个大小为 m x n 的二进制矩阵 grid ,其中 0 表示一个海洋单元格、1 表示一个陆地单元格。
一次 移动 是指从一个陆地单元格走到另一个相邻(上、下、左、右)的陆地单元格或跨过 grid 的边界。
返回网格中 无法 在任意次数的移动中离开网格边界的陆地单元格的数量。
示例 1:
输入:grid = [[0,0,0,0],[1,0,1,0],[0,1,1,0],[0,0,0,0]]
输出:3
解释:有三个 1 被 0 包围。一个 1 没有被包围,因为它在边界上。
示例 2:
输入:grid = [[0,1,1,0],[0,0,1,0],[0,0,1,0],[0,0,0,0]]
输出:0
解释:所有 1 都在边界上或可以到达边界。
提示:
m == grid.length
n == grid[i].length
1 <= m, n <= 500
grid[i][j] 的值为 0 或 1
解题思路
采用深度优先遍历,如果边界能达到的陆地标记为True,由边界不能达到的陆地和海洋标记为True,统计不能有边界达到的陆地数量。
python代码
def numEnclaves(self, grid: List[List[int]]) -> int:
n,m = len(grid),len(grid[0])
vist = [[False]*m for _ in range(n)]
def dfs(x,y):
if x < 0 or y < 0 or x > n-1 or y > m-1 or grid[x][y] == 0 or vist[x][y]:
return
vist[x][y] = True
for dx,dy in (0,1),(0,-1),(-1,0),(1,0):
dfs(x+dx,y+dy)
for i in range(n):
dfs(i,0)
dfs(i,m-1)
for j in range(1,m-1):
dfs(0,j)
dfs(n-1,j)
ans = sum(grid[i][j] and not vist[i][j] for i in range(1, n - 1) for j in range(1, m - 1))
return(ans)