leetcode1034. 边界着色(mid)(每日一题)

力扣链接

题目描述

给你一个大小为 m x n 的整数矩阵 grid ，表示一个网格。另给你三个整数 row、col 和 color 。网格中的每个值表示该位置处的网格块的颜色。

两个网格块属于同一 连通分量 需满足下述全部条件：

• 两个网格块颜色相同
• 在上、下、左、右任意一个方向上相邻
  连通分量的边界 是指连通分量中满足下述条件之一的所有网格块：

在上、下、左、右四个方向上与不属于同一连通分量的网格块相邻
在网格的边界上（第一行/列或最后一行/列）
请你使用指定颜色 color 为所有包含网格块 grid[row][col] 的 连通分量的边界 进行着色，并返回最终的网格 grid 。

示例 1：

输入：grid = [[1,1],[1,2]], row = 0, col = 0, color = 3
输出：[[3,3],[3,2]]

示例 2：

输入：grid = [[1,2,2],[2,3,2]], row = 0, col = 1, color = 3
输出：[[1,3,3],[2,3,3]]

示例 3：

输入：grid = [[1,1,1],[1,1,1],[1,1,1]], row = 1, col = 1, color = 2
输出：[[2,2,2],[2,1,2],[2,2,2]]

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 50
• 1 <= grid[i][j], color <= 1000
• 0 <= row < m
• 0 <= col < n

解题思路

外国小哥题解链接

• dfs

代码(dfs)

class Solution {
    
    
    public int[][] colorBorder(int[][] grid, int row, int col, int color) {
    
    

        int n = grid.length;
        int m = grid[0].length;
        // 如果给的点的值等于要着的颜色 可以直接返回 因为根据题意 只有与给定点的颜色相同才有被判定为边界的资格。一旦颜色相同了 边界的颜色就是color 所以可以直接返回了  但是该条件可要可不要
        if (grid[row][col] == color) {
    
    
            return grid;
        }

        boolean[][] visited = new boolean[n][m];
        dfs(grid, row, col, grid[row][col], visited, color, n, m);

        return grid;

    }

    public void dfs(int[][] grid, int i, int j, int col, boolean[][] visited, int target, int n, int m){
    
    
        // 如果下标越界 或者 与初始点的值不相同 或者 该点被访问过 都不能成为边界
        if (i > n-1 || i < 0 || j > m-1 || j < 0 || grid[i][j] != col || visited[i][j]) {
    
    
            return;
        }

        visited[i][j] = true;
        boolean border = false;

        // 之前已经判断过conditino下标没有越界
        // 其次，如果是grid的边界，那么一定是连分量的边界
        // 最后，如果该点的任意一个方向不为初始点的值，那么该点肯定是位于边界上的，因为如果不是边界，四面的值都应该是val
        if (i == 0 || j == 0 || j == m-1 || i == n-1 || grid[i+1][j] != col || grid[i-1][j] != col || grid[i][j-1] != col || grid[i][j+1] != col) {
    
    
            border = true;
        }

        // dfs四个方向
        dfs(grid, i+1, j, col, visited, target, n, m);
        dfs(grid, i-1, j, col, visited, target, n, m);
        dfs(grid, i, j+1, col, visited, target, n, m);
        dfs(grid, i, j-1, col, visited, target, n, m);

        if (border) {
    
    
            grid[i][j] = target;
        }

    }
}

复杂度

• 时间复杂度: O(mn)
• 空间复杂度: O(mn)