733. 图像渲染

可以理解为图的宽度遍历
下面如果颜色相同就直接退出是防止出现无限递归

class Solution {
public:
class pos
{
	public:
		int x_;
		int y_;
		pos(int x, int y):x_(x),y_(y){}
};

vector<vector<int>> floodFill(vector<vector<int>>& image, int sr, int sc, int color) {
	if(image.empty())
		return image;
	int standard_color = image.at(sr).at(sc);
	if(color == standard_color)
		return image;
	queue<pos>to_use;
	to_use.push(pos(sr,sc));
	image.at(sr).at(sc) = color;
	while(!to_use.empty())
	{
		auto cur_pos = to_use.front();
		to_use.pop();
		if(cur_pos.x_-1 >=0 && image.at(cur_pos.x_-1).at(cur_pos.y_)==standard_color)
		{
			image.at(cur_pos.x_-1).at(cur_pos.y_) = color;
			to_use.push(pos(cur_pos.x_-1, cur_pos.y_));
		}

		if(cur_pos.x_+1 <(int)image.size() && image.at(cur_pos.x_+1).at(cur_pos.y_)==standard_color)
		{
			image.at(cur_pos.x_+1).at(cur_pos.y_) = color;
			to_use.push(pos(cur_pos.x_+1, cur_pos.y_));
		}
		if(cur_pos.y_-1 >=0 && image.at(cur_pos.x_).at(cur_pos.y_-1)==standard_color)
		{
			image.at(cur_pos.x_).at(cur_pos.y_-1) = color;
			to_use.push(pos(cur_pos.x_, cur_pos.y_-1));
		}
		if(cur_pos.y_+1<(int)image[0].size() && image.at(cur_pos.x_).at(cur_pos.y_+1)==standard_color)
		{
			image.at(cur_pos.x_).at(cur_pos.y_+1) = color;
			to_use.push(pos(cur_pos.x_, cur_pos.y_+1));
		}
	}
	return image;
}



};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52