【广度优先遍历-中等】面试题 16.19. 水域大小

原创于 2022-09-10 11:51:57 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#宽度优先 #leetcode #算法

刷题专栏收录该内容

643 篇文章

订阅专栏

这篇博客探讨了两种解决二维数组中湖泊面积问题的方法。方法一是使用递归，通过深度优先搜索（DFS）遍历网格，方法二是利用栈来避免递归调用。这两种方法都用于计算零值（表示水）区域的大小，并最终返回排序后的湖泊面积列表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目】

【代码】
【方法1】递归方法求解
在这里插入图片描述

class Solution:
    def pondSizes(self, land: List[List[int]]) -> List[int]:
        ans=[]
        def dfs(x,y):
            if x>=0 and x<len(land) and y>=0 and y<len(land[0]) and land[x][y]==0:
                land[x][y]=1
                return 1+dfs(x+1,y)+dfs(x-1,y)+dfs(x,y+1)+dfs(x,y-1)+dfs(x-1,y-1)+dfs(x+1,y+1)+dfs(x-1,y+1)+dfs(x+1,y-1)
            return 0

        for i in range(len(land)):
            for j in range(len(land[0])):
                if land[i][j]==0:
                    tmp=dfs(i,j)
                    if tmp:
                        ans.append(tmp)
        ans.sort()
        return ans

【方法2】用栈代替递归
在这里插入图片描述

class Solution:
    def pondSizes(self, land: List[List[int]]) -> List[int]:
        ans=[]
        for i in range(len(land)):
            for j in range(len(land[0])):
                if land[i][j]==0:
                    cnt=0
                    stack=[(i,j)]
                    while stack:
                        x,y=stack.pop()
                        if x>=0 and x<len(land) and y>=0 and y<len(land[0]) and land[x][y]==0:
                            land[x][y]=1
                            cnt+=1
                            stack.append((x+1,y))
                            stack.append((x-1,y))
                            stack.append((x,y+1))
                            stack.append((x,y-1))
                            stack.append((x-1,y-1))
                            stack.append((x+1,y+1))
                            stack.append((x-1,y+1))
                            stack.append((x+1,y-1))
                    if cnt:
                        ans.append(cnt)
        ans.sort()
        return ans