Leetcode--图像渲染

From:https://leetcode-cn.com/problems/flood-fill/

有一幅以二维整数数组表示的图画，每一个整数表示该图画的像素值大小，数值在 0 到 65535 之间。

给你一个坐标 (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，
因为它不是在上下左右四个方向上与初始点相连的像素点。
注意:

image 和 image[0] 的长度在范围 [1, 50] 内。
给出的初始点将满足 0 <= sr < image.length 和 0 <= sc < image[0].length。
image[i][j] 和 newColor 表示的颜色值在范围 [0, 65535]内。

思路

这道题可以直接使用广度优先搜索解决；

使用一个变量来记录(sr, sc)原来的像素颜色便可以原地修改，降低空间复杂度

class Solution {
public:

    vector> floodFill(vector>& image, int sr, int sc, int newColor) {
        // 边界判断
        if(image.size()== 0|| image[0].size()== 0) return image;
        // 如果修改的颜色跟原来的一样则直接返回
        if(image[sr][sc]== newColor) return image;

        // 广度优先搜索的四个方向，分别对应上右下左
        int dx[4]= {-1, 0, 1, 0};
        int dy[4]= {0, 1, 0, -1};

        int row= image.size(), col= image[0].size();
        queue > qu;
        qu.push({sr, sc});

        // (sr,sc)原来的像素颜色
        int originColor= image[sr][sc];
        // 广度优先搜索
        while(!qu.empty()){
            pair temp= qu.front();
            qu.pop();

            image[temp.first][temp.second]= newColor;
            for(int i= 0; i< 4; ++i){
                int x= temp.first+ dx[i], y= temp.second+ dy[i];
                // 找到与原来像素颜色相同的相连像素
                if(0<= x&& x< row&& y>= 0&& y< col
                    && image[x][y]== originColor){
                    qu.push({x, y});
                }
            }
        }
        return image;
    }
};