LeetCode-Python-1034. 边框着色

658 篇文章 23 订阅

给出一个二维整数网格 grid,网格中的每个值表示该位置处的网格块的颜色。

只有当两个网格块的颜色相同,而且在四个方向中任意一个方向上相邻时,它们属于同一连通分量

连通分量的边界是指连通分量中的所有与不在分量中的正方形相邻(四个方向上)的所有正方形,或者在网格的边界上(第一行/列或最后一行/列)的所有正方形。

给出位于 (r0, c0) 的网格块和颜色 color,使用指定颜色 color 为所给网格块的连通分量的边界进行着色,并返回最终的网格 grid 。

 

示例 1:

输入:grid = [[1,1],[1,2]], r0 = 0, c0 = 0, color = 3
输出:[[3, 3], [3, 2]]

示例 2:

输入:grid = [[1,2,2],[2,3,2]], r0 = 0, c0 = 1, color = 3
输出:[[1, 3, 3], [2, 3, 3]]

示例 3:

输入:grid = [[1,1,1],[1,1,1],[1,1,1]], r0 = 1, c0 = 1, color = 2
输出:[[2, 2, 2], [2, 1, 2], [2, 2, 2]]

 

提示:

  1. 1 <= grid.length <= 50
  2. 1 <= grid[0].length <= 50
  3. 1 <= grid[i][j] <= 1000
  4. 0 <= r0 < grid.length
  5. 0 <= c0 < grid[0].length
  6. 1 <= color <= 1000

思路:

BFS往外找,找到符合要求的就变色,用visited记录当前点有没有来过,用record记录原始矩阵(因为过程中会修改原始矩阵)。

 

class Solution(object):
    def colorBorder(self, grid, r0, c0, color):
        """
        :type grid: List[List[int]]
        :type r0: int
        :type c0: int
        :type color: int
        :rtype: List[List[int]]
        """

        m= len(grid)
        if not m:
            return grid
        n = len(grid[0])
        if not n:
            return grid
        
        dx = [1, -1, 0, 0]
        dy = [0, 0, -1, 1]
        originalcolor = grid[r0][c0]
        record = list()
        visited = list()
        for row in grid:
            record.append(row[:])
            visited.append(row[:])
            
        def checkNeibor(x, y):     
            #返回4说明不染色,返回其他说明满足染色条件
            if x in [0, m - 1] or y in [0, n - 1]:#在边界上
                return -1
            cnt = 0
            for j in range(4):
                xx = x + dx[j]
                yy = y + dy[j]
                if 0 <= xx < m and 0<= yy < n and originalcolor == record[xx][yy]:
                    cnt += 1
            return cnt 
  
        if checkNeibor(r0, c0) != 4:
            grid[r0][c0] = color
        
        queue = [[r0,c0]]
        while(queue):
            newqueue = list()
            for node in queue:
                x0 = node[0]
                y0 = node[1]
                for k in range(4):
                    x = x0 + dx[k]
                    y = y0 + dy[k]

                    if 0 <= x < m and 0<= y < n and originalcolor == record[x][y] and visited[x][y] != -1:
                        newqueue.append([x, y])
                        visited[x][y] = -1
                        # print [x, y], queue
                        if checkNeibor(x, y) != 4:
                            grid[x][y] = color           
                        
            queue = newqueue[:]
        
        return grid

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值