深度优先遍历--最大的岛屿

时间:2022-07-23
本文章向大家介绍深度优先遍历--最大的岛屿,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

问题描述:给定一个二维矩阵,0表示水,1表示陆地,一个岛屿是指相邻的上下左右的陆地面积,求最大的岛屿

a=[[1,1,1,0,0,0],
   [1,1,1,0,0,0],
   [1,0,0,0,1,1],
   [0,1,1,0,1,0],
   [0,1,1,0,0,0]]
area = 0
def maxAreaOfIsland(a):
   #记录地图的行,列 
    row=len(a)
    col=len(a[0])
    for i in range(row):
        for j in range(col):
            if a[i][j]==1:
                #存储当前岛屿的面积
                cur=1
                #深度优先遍历
                dfs(i,j,cur,a)
    return area
def dfs(i,j,cur,a):
    #定义全局变量
    global area
    #将以已经遍历过的标记
    a[i][j]=2
    if i>0 and a[i-1][j]==1:
        cur=dfs(i-1,j,cur+1,a)
    if i<len(a)-1 and a[i+1][j]==1:
        cur = dfs(i+1,j,cur+1,a)
    if j>0 and a[i][j-1]==1:
        cur=dfs(i,j-1,cur+1,a)
    if j<len(a[0])-1 and a[i][j+1]==1:
        cur=dfs(i,j+1,cur+1,a)
    #更新最大面积
    area=max(area,cur)
    return cur
print(maxAreaOfIsland(a))
输出:7            

另一种写法:

a=[[1,1,1,0,0,0],
   [1,1,1,0,0,0],
   [1,0,0,0,1,1],
   [0,1,1,0,1,0],
   [0,1,1,0,0,0]]

def maxAreaOfIsland(a):
    global cur
    area = 0
    row=len(a)
    col=len(a[0])
    for i in range(row):
        for j in range(col):
            cur=0
            dfs(i,j,a)
            area=max(cur,area)
    return area

def dfs(i,j,a):
    global cur
    if i<0 or i>len(a)-1 or j<0 or j>len(a[0])-1 or a[i][j]!=1:
        return
    cur+=1
    a[i][j]=2
    dfs(i-1,j,a)
    dfs(i+1,j,a)
    dfs(i,j-1,a)
    dfs(i,j+1,a)
print(maxAreaOfIsland(a))
输出:7