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