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

最新推荐文章于 2024-10-07 07:09:00 发布

菜菜2022

最新推荐文章于 2024-10-07 07:09:00 发布

阅读量109

点赞数

分类专栏：刷题文章标签：宽度优先 leetcode 算法

本文链接：https://blog.csdn.net/kz_java/article/details/126795159

版权

刷题专栏收录该内容

643 篇文章 5 订阅

订阅专栏

【题目】

【代码】
【方法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