Java——图像渲染

题目链接

leetcode在线oj题——图像渲染

题目描述

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

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

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

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

题目示例

输入: 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，因为它不是在上下左右四个方向上与初始点相连的像素点。

输入: 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

解题思路

可以使用深度优先的遍历，得出结果

定义一个nextposition的数组，代表接下来要走的位置
定义一个isUsed布尔数组，记录当前位置是否被遍历过

定义DFS方法，参数分别是image数组，isUsed数组，数组行数，数组列数，当前行，当前列，要更新的颜色，当前位置颜色

先修改当前位置的颜色，以及标记该位置为已遍历

然后分别找到下一个要遍历的位置，如果这个位置是已经遍历过或者越界了，就不做任何操作，否则继续递归DFS函数

最后返回image数组即可

代码

class Solution {
    int[][] nextPositoin = {{1, 0}, {-1, 0}, {0, -1}, {0, 1}};
    public int[][] floodFill(int[][] image, int sr, int sc, int color) {
        if(image == null){
            return image;
        }
        boolean[][] isUsed = new boolean[image.length][image[0].length];
        int oldColor = image[sr][sc];
        DFS(image, isUsed, image.length, image[0].length, sr, sc, color, oldColor);
        return image;
    }

    public void DFS(int[][] image, boolean[][] isUsed, int row, int col,
                    int curX, int curY, int newColor, int oldColor){
        //当前位置修改颜色并标记已遍历
        image[curX][curY] = newColor;
        isUsed[curX][curY] = true;

        //确定下一个位置
        for (int i = 0; i < 4; i++) {
            int newX = curX + nextPositoin[i][0];
            int newY = curY + nextPositoin[i][1];

            //判断是否越界
            if(newX >= row || newX < 0 || newY >= col || newY < 0){
                continue;
            }

            //判断是否需要修改颜色
            if(image[newX][newY] == oldColor && !isUsed[newX][newY]){
                DFS(image, isUsed, row, col, newX, newY, newColor, oldColor);
            }
        }
    }
}