【深度优先搜索-中等】1034. 边界着色

最新推荐文章于 2025-11-30 20:35:27 发布

原创最新推荐文章于 2025-11-30 20:35:27 发布 · 166 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法 #leetcode

刷题专栏收录该内容

643 篇文章

订阅专栏

这篇博客介绍了两种方法实现联通分量边界的着色，主要涉及图论和深度优先搜索（DFS）算法。代码展示了如何遍历并更新二维网格中特定颜色区域的边界，确保只改变边界上的元素。第一种方法使用了广度优先搜索（BFS），第二种方法则采用DFS。这两种方法都有效地解决了题目所给的问题。

【题目】
在这里插入图片描述注意：这题是对联通分量的边界进行着色，笔者第一遍做的时候就是忽略了这一点

【代码】
在这里插入图片描述

class Solution:
    def colorBorder(self, grid: List[List[int]], row: int, col: int, color: int) -> List[List[int]]:
        origin=grid[row][col]
        if origin==color:return grid
        stack=[(row,col)]
        p=set()
        while stack:
            row,col=stack.pop()
            if 0<=row<len(grid) and 0<=col<len(grid[0]) and grid[row][col]==origin:
                p.add((row,col))
                for mx,my in [(row-1,col),(row+1,col),(row,col-1),(row,col+1)]:
                    if (mx,my) not in p:
                        stack.append((mx,my))
        for x,y in p:            
            for row,col in [(x-1,y),(x+1,y),(x,y+1),(x,y-1)]:
                if (0<=row<len(grid) and 0<=col<len(grid[0]) and grid[row][col]!=origin and (row,col) not in p) or not(0<=row<len(grid) and 0<=col<len(grid[0])):
                    grid[x][y]=color
        return grid

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

class Solution:
    def colorBorder(self, grid: List[List[int]], row: int, col: int, color: int) -> List[List[int]]:
        origin=grid[row][col]
        if origin==color:return grid
        def dfs(row,col):
            if 0<=row<len(grid) and 0<=col<len(grid[0]) and grid[row][col]==origin:
                grid[row][col]=99999
                return {(row,col)}|dfs(row-1,col)|dfs(row+1,col)|dfs(row,col-1)|dfs(row,col+1)
            return set()
        p=dfs(row,col)
        for x,y in p:
            for row,col in [(x-1,y),(x+1,y),(x,y+1),(x,y-1)]:
                if (0<=row<len(grid) and 0<=col<len(grid[0]) and grid[row][col]!=origin and (row,col) not in p) or not(0<=row<len(grid) and 0<=col<len(grid[0])):
                    grid[x][y]=color
                elif (0<=row<len(grid) and 0<=col<len(grid[0]) and grid[row][col]==99999):
                    grid[row][col]=origin
        return grid