733. 图像渲染

DFS:

733. 图像渲染

有一幅以 m x n 的二维整数数组表示的图画 image ，其中 image[i][j] 表示该图画的像素值大小。

你也被给予三个整数 sr , sc 和 newColor 。你应该从像素 image[sr][sc] 开始对图像进行 上色填充 。

为了完成 上色工作 ，从初始像素开始，记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点，接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点，……，重复该过程。将所有有记录的像素点的颜色值改为 newColor 。

最后返回 经过上色渲染后的图像 。

示例 1:

输入: image = [[1,1,1],[1,1,0],[1,0,1]]，sr = 1, sc = 1, newColor = 2
输出: [[2,2,2],[2,2,0],[2,0,1]]
解析: 在图像的正中间，(坐标(sr,sc)=(1,1)),在路径上所有符合条件的像素点的颜色都被更改成2。
注意，右下角的像素没有更改为2，因为它不是在上下左右四个方向上与初始点相连的像素点。

示例 2:

输入: image = [[0,0,0],[0,0,0]], sr = 0, sc = 0, newColor = 2
输出: [[2,2,2],[2,2,2]]

提示:

• m == image.length
• n == image[i].length
• 1 <= m, n <= 50
• 0 <= image[i][j], newColor < 216
• 0 <= sr < m
• 0 <= sc < n

思路及算法

我们从给定的起点开始，进行广度优先搜索。每次搜索到一个方格时，如果其与初始位置的方格颜色相同，就将该方格加入队列，并将该方格的颜色更新，以防止重复入队。

代码

class Solution {
    public int[][] floodFill(int[][] image, int sr, int sc, int newColor) {
        //BFS：
        //先将坐标image[sr][sc]染色作为第一层,然后对其上下左右染色作为第二层
        //再将第二层的分别的上下左右染色,作为第三层,以此类推

        //定义移动上下左右四种情况
        int[] dx = new int[]{1, 0, 0, -1};
        int[] dy = new int[]{0, 1, -1, 0};

        //矩阵行列边界
        int row = image.length;
        int col = image[0].length;

        //判断初始颜色相同时返回原图,因为在本次染色前,原图已经符合条件
        int currColor = image[sr][sc];
        if (currColor == newColor) {
            return image;
        }

        //创建队列存储每一次要染色的方格x,y坐标
        Queue<int[]> queue = new LinkedList<>();
        //先将起始点染色
        queue.offer(new int[]{sr, sc});
        image[sr][sc] = newColor;

        //BFS搜索
        //外层循环改变起始坐标
        //外层循环判断条件,当队列为空,则涂色完成
        while (!queue.isEmpty()) {

            //分解坐标
            int cell[] = queue.poll();
            int x = cell[0];
            int y = cell[1];

            //内层循环改变朝4个方向移动
            //坐标移动判断条件是x,y坐标移动均在矩阵内,且移动的下一个方格颜色与当前(染色前)方格相同;
            for (int i = 0; i < 4; i++) {
                //每一次改变起始坐标
                int mx = x + dx[i];
                int my = y + dy[i];

                if (mx >= 0 && mx < row && my >= 0 && my < col && image[mx][my] == currColor) {
                    image[mx][my] = newColor;
                    queue.offer(new int[]{mx, my});
                }
            }
        }

        return image;
    }
}

记录：2022年12月19日~