LeetCode-695、岛屿的最大面积-中等
给定一个包含了一些 0 和 1的非空二维数组 grid , 一个 岛屿 是由四个方向 (水平或垂直) 的 1 (代表土地) 构成的组合。你可以假设二维矩阵的四个边缘都被水包围着。
找到给定的二维数组中最大的岛屿面积。(如果没有岛屿,则返回面积为0。)
示例 1:
[[0,0,1,0,0,0,0,1,0,0,0,0,0],
[0,0,0,0,0,0,0,1,1,1,0,0,0],
[0,1,1,0,1,0,0,0,0,0,0,0,0],
[0,1,0,0,1,1,0,0,1,0,1,0,0],
[0,1,0,0,1,1,0,0,1,1,1,0,0],
[0,0,0,0,0,0,0,0,0,0,1,0,0],
[0,0,0,0,0,0,0,1,1,1,0,0,0],
[0,0,0,0,0,0,0,1,1,0,0,0,0]]
对于上面这个给定矩阵应返回 6。注意答案不应该是11,因为岛屿只能包含水平或垂直的四个方向的‘1’。
示例 2:
[[0,0,0,0,0,0,0,0]]
对于上面这个给定的矩阵, 返回 0。
注意:给定的矩阵grid 的长度和宽度都不超过 50。
代码:DFS
class Solution:
def maxAreaOfIsland(self, grid: List[List[int]]) -> int:
cur = 0
if not grid or not grid[0]:
return cur
def func(i, j):
grid[i][j] = 0
stack = [(i-1, j), (i+1, j), (i, j-1), (i, j+1)]
tmp = 1
for (x, y) in stack:
if row > x >= 0 and col > y >= 0:
if grid[x][y] == 1:
tmp += func(x, y)
else:
return tmp
row = len(grid)
col = len(grid[0])
for i in range(row):
for j in range(col):
if grid[i][j] == 1:
cur = max(cur, func(i, j))
return cur