【深度优先搜索-简单】733. 图像渲染、面试题 08.10. 颜色填充

原创已于 2022-09-14 12:12:30 修改 · 191 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法 #leetcode

于 2022-09-14 12:08:46 首次发布

刷题专栏收录该内容

643 篇文章

订阅专栏

【题目】
在这里插入图片描述
【代码】
【方法1】栈模拟深度优先遍历
就是用栈代替dfs，注意这里如果image[sr][sc]==color程序直接结束，返回原始的image
因为图像渲染就是将原始颜色渲染成新颜色，如果前后颜色相同就没必要渲染，如果不单独判断前后颜色是否相同，直接对图像进行颜色变换，会导致死循环，如果不想判断，那必须记录已经访问过的节点的信息，不重复访问已经访问的节点
在这里插入图片描述

class Solution:
    def floodFill(self, image: List[List[int]], sr: int, sc: int, color: int) -> List[List[int]]:
        stack=[(sr,sc)]
        origin=image[sr][sc]
        if origin==color:
            return image
        while stack:
            sr,sc=stack.pop()
            if 0<=sr<len(image) and 0<=sc<len(image[0]):
                if image[sr][sc]==origin:
                    image[sr][sc]=color
                    stack.append((sr-1,sc))
                    stack.append((sr+1,sc))
                    stack.append((sr,sc+1))
                    stack.append((sr,sc-1))
        return image

【方法2】dfs
在这里插入图片描述

class Solution:
    def floodFill(self, image: List[List[int]], sr: int, sc: int, color: int) -> List[List[int]]:
        stack=[(sr,sc)]
        origin=image[sr][sc]
        if origin==color:
            return image
        def dfs(sr,sc):
            if 0<=sr<len(image) and 0<=sc<len(image[0]):
                if image[sr][sc]==origin:
                    image[sr][sc]=color
                    dfs(sr-1,sc)
                    dfs(sr+1,sc)
                    dfs(sr,sc+1)
                    dfs(sr,sc-1)
        dfs(sr,sc)
        return image

【方法4】
在这里插入图片描述

class Solution:
    def floodFill(self, image: List[List[int]], sr: int, sc: int, color: int) -> List[List[int]]:
        queue=[(sr,sc)]
        origin=image[sr][sc]
        if origin==color:
            return image
        while queue:
            sr,sc=queue.pop(0)
            if 0<=sr<len(image) and 0<=sc<len(image[0]):
                if image[sr][sc]==origin:
                    image[sr][sc]=color
                    queue.append((sr-1,sc))
                    queue.append((sr+1,sc))
                    queue.append((sr,sc+1))
                    queue.append((sr,sc-1))

        return image