leetcode每日一题 python解法 3月15日 深度优先

2020-03-15  本文已影响0人  Never肥宅

大家315快乐!!!(好像怪怪的噢)

难度:中等

题目内容:

给定一个包含了一些 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。

题解:

这题看上去要找到所有的岛屿比大小,岛屿的话就是连成一片的1。
所以写个递归让他能返回一整片岛屿的面积,并且把岛屿的1清零。
然后遍历每一个点,如果是1就递归算一下面积
把所有的面积的最大值return即可
这个方法专业点讲就是DFS,深度优先搜索。
好了开写

class Solution:
    def maxAreaOfIsland(self, grid: List[List[int]]) -> int:
        islandArea = [0]#没有就输出0
        for row in range(len(grid)):
            for column in range(len(grid[0])):
                if (grid[row][column] == 1):
                    islandArea.append(self.countArea(grid,row,column))
        return max(islandArea)

    def countArea(self,grid,x,y):
        if x<0 or y < 0 or x >= len(grid) or y >= len(grid[0]):#出界了
            return 0
        if grid[x][y] == 1:
            grid[x][y] = 0 #防止被重复计算
            return 1 + self.countArea(grid,x+1,y) +\
                self.countArea(grid,x,y+1) +\
                self.countArea(grid,x,y-1) +\
                self.countArea(grid,x-1,y)
        if grid[x][y] == 0:
            return 0

效果还行吧


image.png
上一篇下一篇

猜你喜欢

热点阅读