题目链接

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);
            }
        }
    }
}