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，不过这次保存每个岛屿的最大面积。需要注意的地方是，每次判断周边结点是否能入栈时，如果能入栈，则直接改变值，这样可以避免1个结点重复入栈。

代码

class Solution:
    def maxAreaOfIsland(self, grid: List[List[int]]) -> int:
        m, n = len(grid), len(grid[0])
        res = 0
        def dfs(row: int, col: int) -> int:
            stack = [(row, col)]
            grid[row][col] = -1
            area = 0
            while stack:
                x, y = stack.pop()
                area += 1
                for new_x, new_y in zip((x, x, x + 1, x - 1), (y + 1, y - 1, y, y)):
                    if 0 <= new_x < m and 0 <= new_y < n and grid[new_x][new_y] == 1:
                        stack.append((new_x, new_y))
                        grid[new_x][new_y] = -1
            return area
        for row in range(m):
            for col in range(n):
                if grid[row][col] == 1:
                    res = max(res, dfs(row, col))
        return res