有一幅以 m x n
的二维整数数组表示的图画 image
,其中 image[i][j]
表示该图画的像素值大小。
你也被给予三个整数 sr
, sc
和 newColor
。你应该从像素 image[sr][sc]
开始对图像进行 上色填充 。
为了完成 上色工作 ,从初始像素开始,记录初始坐标的 上下左右四个方向上 像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应 四个方向上 像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为 newColor
。
最后返回 经过上色渲染后的图像 。
class Solution
{
constexpr static int x[4] = {0,1,0,-1};//数字移动比对表
constexpr static int y[4] = {1,0,-1,0};
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color)
{
int m = image.size();
int n = image[0].size();
ret = image[sr][sc];
if(ret == color)//如果颜色一致,就直接返回
{
return image;
}
image[sr][sc] = color;
queue<pair<int,int>> open;//pair类似于结构体,但是只能定义需要两个属性的,类型可以不同
open.emplace(sr,sc);//用emplace不用重新定义一个类,emplace会自动调用其对应类型的构造函数,如果用push要定义一个类
while(!open.empty())
{
int tx = open.front().first;//pair的函数,访问第一个属性
int ty = open.front().second;//pair的函数,访问第二个属性
open.pop();
for(int k = 0;k < 4;k++)
{
int mx = tx + x[k];
int my = ty + y[k];
if(mx >= 0 && mx < m && my >= 0 && my < n&& image[mx][my] == ret)
{
open.emplace(mx,my);
image[mx][my] = color;
}
}
}
return image;
}
public:
int ret = 0;//记录原来的颜色
};