美文网首页
leetcode每日一题 python解法 3月15日 深度优先

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

作者: Never肥宅 | 来源:发表于2020-03-15 01:15 被阅读0次

    大家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

    相关文章

      网友评论

          本文标题:leetcode每日一题 python解法 3月15日 深度优先

          本文链接:https://www.haomeiwen.com/subject/pbpashtx.html