树的深度优先搜索--最大的岛屿

问题描述：给定一个二维矩阵，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