leetcode -695 Max Area of Island

Given a non-empty 2D array grid of 0's and 1's, an island is a group of 1's (representing land) connected 4-directionally (horizontal or vertical.) You may assume all four edges of the grid are surrounded by water.

Find the maximum area of an island in the given 2D array. (If there is no island, the maximum area is 0.)

input:
[[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]]
output:
6

这道题就是一个简单的搜索问题，代码比较简单。

class Solution:
    def maxAreaOfIsland(self, grid: List[List[int]]) -> int:
        maxIsland = 0
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                if grid[i][j] == 1:
                    grid[i][j] = 0
                    stack = []
                    count = 0
                    stack.append((i, j))
                    while len(stack) != 0:
                        (x, y) = stack.pop()
                        count += 1
                        if x - 1 >= 0 and grid[x - 1][y] == 1:
                            stack.append((x - 1, y))
                            grid[x - 1][y] = 0
                        if x + 1 < len(grid) and grid[x + 1][y] == 1:
                            stack.append((x + 1, y))
                            grid[x + 1][y] = 0
                        if y - 1 >= 0 and grid[x][y - 1] == 1:
                            stack.append((x, y - 1))
                            grid[x][y - 1] = 0
                        if y + 1 < len(grid[0]) and grid[x][y + 1] == 1:
                            stack.append((x, y + 1))
                            grid[x][y + 1] = 0
                    if count > maxIsland:
                        maxIsland = count
        return maxIsland