LeetCode-695、岛屿的最大面积-中等

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