有一幅以二维整数数组表示的图画,每一个整数表示该图画的像素值大小,数值在 0 到 65535 之间。
为了完成上色工作,从初始坐标开始,记录初始坐标的上下左右四个方向上像素值与初始坐标相同的相连像素点,接着再记录这四个方向上符合条件的像素点与他们对应四个方向上像素值与初始坐标相同的相连像素点,……,重复该过程。将所有有记录的像素点的颜色值改为新的颜色值。
示例 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]内。
class Solution {
public:
vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int newColor) {
int oldcolor=image[sr][sc];
if(oldcolor==newColor) return image;
image[sr][sc]=newColor;
vector<int >tmp{sr,sc};
stack<vector<int>> pos;
pos.push(tmp);
vector<vector<int>>step={{-1,0},{0,-1},{1,0},{0,1}};
while(!pos.empty())
{
vector<int >t = pos.top();
pos.pop();
for(auto s: step)
{
int i=t[0]+s[0];
int j=t[1]+s[1];
if(i>=0 && i<image.size() && j >=0 && j<image[0].size() && image[i][j]==oldcolor)
{
vector<int > ind={i,j};
pos.push(ind);
image[i][j]=newColor;
// cout<< i<<" "<<j<<" "<< image.size()<<endl;
}
}
}
return image;
}
};
执行用时 : 36 ms, 在Flood Fill的C++提交中击败了95.63% 的用户
内存消耗 : 10.9 MB, 在Flood Fill的C++提交中击败了70.62% 的用户